mindspore.ops.Concat
- class mindspore.ops.Concat(axis=0)[源代码]
在指定轴上拼接输入Tensor。
更多参考详见
mindspore.ops.concat()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> input_x1 = Tensor(np.array([[0, 1], [2, 1]]).astype(np.float32)) >>> input_x2 = Tensor(np.array([[0, 1], [2, 1]]).astype(np.float32)) >>> op = ops.Concat() >>> output = op((input_x1, input_x2)) >>> print(output) [[0. 1.] [2. 1.] [0. 1.] [2. 1.]] >>> op = ops.Concat(1) >>> output = op((input_x1, input_x2)) >>> print(output) [[0. 1. 0. 1.] [2. 1. 2. 1.]]