mindspore.ops.hsplit
- mindspore.ops.hsplit(input, indices_or_sections)[源代码]
水平地将输入Tensor分割成多个子Tensor。等同于 \(axis=1\) 时的 ops.tensor_split 。
- 参数:
input (Tensor) - 待分割的Tensor。
indices_or_sections (Union[int, tuple(int), list(int)]) - 参考
mindspore.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]]))