megengine.functional.stack¶

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

在一个新的轴上连接一个序列中的张量。输入张量必须有相同的形状。

参数

inps – 输入张量。
axis – 用来连接的轴。
device – 输出所在的计算节点。默认：None

返回

输入拼接好的张量。

实际案例

>>> import numpy as np
>>> x1 = Tensor(np.arange(0, 3, dtype=np.float32).reshape((3)))
>>> x2 = Tensor(np.arange(6, 9, dtype=np.float32).reshape((3)))
>>> out = F.stack([x1, x2], axis=0)
>>> out.numpy()
array([[0., 1., 2.],
       [6., 7., 8.]], dtype=float32)

megengine.functional.concat

megengine.functional.split