mindspore.numpy.vsplit

mindspore.numpy.vsplit(x, indices_or_sections)

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

参数：

• 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(9).reshape((3, 3)).astype('float32')
>>> output = np.vsplit(input_x, 3)
>>> print(output)
(Tensor(shape=[1, 3], dtype=Float32,
  value=[[ 0.00000000e+00,  1.00000000e+00,  2.00000000e+00]]),
 Tensor(shape=[1, 3], dtype=Float32,
  value=[[ 3.00000000e+00,  4.00000000e+00,  5.00000000e+00]]),
 Tensor(shape=[1, 3], dtype=Float32,
  value=[[ 6.00000000e+00,  7.00000000e+00,  8.00000000e+00]]))