mindspore.numpy.hsplit

mindspore.numpy.hsplit(x, indices_or_sections)

水平（按列）将Tensor分割为多个sub-tensor。此操作等同于使用 \(axis=1\) （默认值）进行分割，不论数组的维度如何，始终沿第二轴分割。

参数：

• x (Tensor) - 待分割的Tensor。

• indices_or_sections (Union[int, tuple(int), list(int)]) - 如果是整数 \(N\) ，Tensor将沿轴分割为 \(N\) 个相等的sub-tensor。如果是tuple(int)、list(int)或排序后的整数，则指示沿轴的分割位置。例如，对于 \(axis=0\) ， \([2,3]\) 将产生三个sub-tensor： \(x[:2]\) 、 \(x[2:3]\) 和 \(x[3:]\) 。如果索引超出轴上数组的维度，则相应地返回空子数组。

返回：

sub-tensor列表。

异常：

• TypeError - 如果参数 indices_or_sections 不是整数。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> input_x = np.arange(6).reshape((2, 3)).astype('float32')
>>> output = np.hsplit(input_x, 3)
>>> print(output)
(Tensor(shape=[2, 1], dtype=Float32,
value=[[ 0.00000000e+00],
       [ 3.00000000e+00]]),
Tensor(shape=[2, 1], dtype=Float32,
value=[[ 1.00000000e+00],
       [ 4.00000000e+00]]),
Tensor(shape=[2, 1], dtype=Float32,
value=[[ 2.00000000e+00],
       [ 5.00000000e+00]]))