BatchConcat
函数功能
将多个Tensor进行组batch,按照batch维组装,默认输入的每个Tensor第一维为batch维,内存连续。
函数原型
friend APP_ERROR BatchConcat(const std::vector<Tensor> &inputs, Tensor &output)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
inputs |
输入 |
待组batch的Tensor列表。 |
output |
输出 |
组装好batch的Tensor。 |
返回参数说明
数据结构 |
说明 |
---|---|
APP_ERROR |
程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: Tensor