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.]]

megengine.functional.squeeze

megengine.functional.stack