BatchVector
函数功能
合并batch组合成一个Tensor。该函数实际调用BatchConcat和BatchStack，实现组batch操作，通过keepDims参数控制具体调用函数。
函数原型
static APP_ERROR BatchVector(const std::vector<TensorBase> &inputs, TensorBase &output, const bool &keepDims = false);
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
inputs |
输入 |
多batch模型，将TensorBase对象列表作为输出。列表大小不超过1024。 |
output |
输出 |
组batch后输出一个TensorBase对象。 |
keepDims |
输入 |
保持其多维特性的参数。
|
返回参数说明
数据结构 |
说明 |
|---|---|
APP_ERROR |
程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase