batch_concat

函数功能

将多个Tensor进行组batch,按照第0维组装,内存连续。

函数原型

1
batch_concat(inputs: List)

输入参数说明

参数名

类型

说明

inputs

List[base.Tensor]

Tensor列表。

返回参数说明

Tensor对象。

抛异常接口

传入Tensor异常、组Tensor后输出的Tensor异常,抛出runtime异常。

支持的型号

Atlas 200/300/500 推理产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品