mindspore.numpy.concatenate

查看源文件
mindspore.numpy.concatenate(arrays, axis=0)[源代码]

沿现有轴连接一系列Tensor。

说明

为了匹配Numpy的行为,当 \(axis >= 32\) 时不会引发值错误, axis 将被视为 None

参数:
  • arrays (Union[Tensor, tuple(Tensor), list(Tensor)]) - 要连接的一个Tensor或Tensor列表。

  • axis (Union[None, int], 可选) - 沿该轴连接Tensor。如果 axisNone ,则在使用前会先将Tensor展平。默认值: 0

返回:

从一个Tensor或Tensor列表中连接后的Tensor。

异常:
  • TypeError - 如果输入参数非上述给定的类型。

  • ValueError - 如果 axis 不在范围 \([-ndim, ndim-1]\) 内,且小于32。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> x1 = np.ones((1,2,3))
>>> x2 = np.ones((1,2,1))
>>> x = np.concatenate((x1, x2), axis=-1)
>>> print(x.shape)
(1, 2, 4)