BatchStack
函数功能
将多batch的Tensor组合成一个Tensor,相比于BatchConcat拓展了Tensor的维度。
函数原型
static APP_ERROR TensorBase::BatchStack(const std::vector<TensorBase> &inputs, TensorBase &output);
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
inputs |
输入 |
Tensor列表,列表大小不超过1024。 |
output |
输出 |
组合后的Tensor对象。 |
返回参数说明
数据结构 |
说明 |
---|---|
APP_ERROR |
程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: TensorBase