mindspore.ops.vsplit

mindspore.ops.vsplit(input, indices_or_sections)

Split the input tensor with two or more dimensions, into multiple sub-tensors vertically according to indices_or_sections.

It is equivalent to ops.tensor_split with $axis=0$ .

Parameters

• input (Tensor) – The input tensor.

• indices_or_sections (Union[int, tuple(int), list(int)]) – See indices_or_sections argument in mindspore.ops.tensor_split().

Returns

Tuple of tensors.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> input = mindspore.tensor([[ 0,  1,  2,  3],
...                           [ 4,  5,  6,  7],
...                           [ 8,  9, 10, 11],
...                           [12, 13, 14, 15]])
>>> mindspore.ops.vsplit(input, 2)
(Tensor(shape=[2, 4], dtype=Int64, value=
 [[0, 1, 2, 3],
  [4, 5, 6, 7]]),
 Tensor(shape=[2, 4], dtype=Int64, value=
 [[ 8,  9, 10, 11],
  [12, 13, 14, 15]]))
>>> mindspore.ops.vsplit(input, [3, 6])
(Tensor(shape=[3, 4], dtype=Int64, value=
 [[ 0,  1,  2,  3],
  [ 4,  5,  6,  7],
  [ 8,  9, 10, 11]]),
 Tensor(shape=[1, 4], dtype=Int64, value=
 [[12, 13, 14, 15]]),
 Tensor(shape=[0, 4], dtype=Int64, value=
 ))