mindspore.ops.hsplit

查看源文件
mindspore.ops.hsplit(input, indices_or_sections)[源代码]

水平地将输入Tensor分割成多个子Tensor。等同于 \(axis=1\) 时的 ops.tensor_split

参数:
返回:

tuple[Tensor]。

异常:
  • TypeError - 如果 input 不是Tensor。

  • ValueError - 如果 input 的维度小于2。

支持平台:

Ascend GPU CPU

样例:

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