- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
将输入的多个Tensor沿指定维度(dim)拼接,返回一个拼接后的Tensor。
[object Object]
返回输出Tensor,Tensor的数据类型和tensors的任一tensor数据类型相同,Shape与tensors任一tensor相同(除dim对应维度),dim对应维度为tensors各个tensor对应维度之和。
1.源操作数tensors的大小需要大于等于2,即len(tensors )>=2;小于等于128。(支持输入一个tensor情况,精度暂时不保证);
2.输入tensor数据类型相同、维度数量相同,并且除待拼接维度(dim)之外的每个维度值相同,待拼接维度validShape等于tensor对应维度值,其余维度tensors validShape相同;
3.dim: -input.dim <= dim < input.dim(input对应tensors的任一tensor);
4.设置viewshape时,dim对应维度不切块(即viewshape对应值>=tensors任一tensor的对应值);
5.输出Tensor的validShape需由用户在调用concat前确保正确,该接口不会自动推导。
6.Tensor类型输入不支持[object Object]格式。
调用该operation接口前,应通过set_vec_tile_shapes设置TileShape。
TileShape维度应和输出一致。
如输入tensors维度为[m, c1, p],[m, c2, p],输出为[m, c1+c2, p],TileShape设置为[m1, n1, p1],则m1, p1分别用于切分m, p轴,n1用于切分c1和c2轴。
[object Object]
[object Object]
结果示例如下:
[object Object]