下载
中文
注册

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说明