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