mindspore.ops.chunk

查看源文件
mindspore.ops.chunk(input, chunks, axis=0)[源代码]

沿着指定轴 axis 将输入Tensor切分成 chunks 个sub-tensor。

说明

此函数返回的数量可能小于通过 chunks 指定的数量!

参数:
  • input (Tensor) - 被切分的Tensor。

  • chunks (int) - 要切分的sub-tensor数量。

  • axis (int,可选) - 指定需要分割的维度。默认值: 0

返回:

tuple[Tensor]。

异常:
  • TypeError - input 不是Tensor。

  • TypeError - axis 不是int类型。

  • TypeError - chunks 不是int。

  • ValueError - 参数 axis 超出 \([-input.ndim, input.ndim)\) 范围。

  • ValueError - 参数 chunks 不是正数。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import ops, Tensor
>>> input_x = np.arange(9).astype("float32")
>>> output = ops.chunk(Tensor(input_x), 3)
>>> print(output)
(Tensor(shape=[3], dtype=Float32, value= [ 0.00000000e+00,  1.00000000e+00,  2.00000000e+00]),
 Tensor(shape=[3], dtype=Float32, value= [ 3.00000000e+00,  4.00000000e+00,  5.00000000e+00]),
 Tensor(shape=[3], dtype=Float32, value= [ 6.00000000e+00,  7.00000000e+00,  8.00000000e+00]))