megengine.functional.concat

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

拼接一些张量

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

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

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

返回类型

Tensor

返回

输出张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

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])
print(out.numpy())

输出:

[[ 0.  1.  2.]
 [ 3.  4.  5.]
 [ 6.  7.  8.]
 [ 9. 10. 11.]]