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)