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)

megengine.functional.squeeze

megengine.functional.stack