megengine.functional.concat

concat(inps, axis=0, device=None)[源代码]

拼接一些张量

参数
  • inps (Iterable[Tensor]) – 将要进行拼接的输入张量。

  • axis (int) – 对张量进行拼接的维度。默认:0

  • device – 输出所在的计算节点。 默认:None

返回类型

Tensor

返回

输出张量。

实际案例

>>> import numpy as np
>>> data1 = Tensor(np.arange(0, 6, dtype=np.float32).reshape((2, 3)))
>>> data2 = Tensor(np.arange(6, 12, dtype=np.float32).reshape((2, 3)))
>>> out = F.concat([data1, data2])
>>> out.numpy()
array([[ 0.,  1.,  2.],
       [ 3.,  4.,  5.],
       [ 6.,  7.,  8.],
       [ 9., 10., 11.]], dtype=float32)