mindspore.ops.hsplit

mindspore.ops.hsplit(input, indices_or_sections)[源代码]

将输入tensor水平分割成多个子tensor。等同于 $a x i s = 1$ 时的 ops.tensor_split 。

参数：

input (Tensor) - 输入tensor。
indices_or_sections (Union[int, tuple(int), list(int)]) - 参考 mindspore.ops.tensor_split()。

返回：

由多个tensor组成的tuple。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> input_x = np.arange(6).reshape((2, 3)).astype('float32')
>>> output = mindspore.ops.hsplit(mindspore.tensor(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]]))