mindspore.Tensor.chunk

mindspore.Tensor.chunk(chunks, dim=0)

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

说明

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

警告

这是一个实验性API，后续可能修改或删除。

参数：

chunks (int) - 要切分的sub-tensor数量。
dim (int，可选) - 指定需要分割的维度。默认值： 0 。

返回：

tuple[Tensor]。

异常：

TypeError - dim 不是int类型。
TypeError - chunks 不是int。
ValueError - 参数 dim 超出 \([-self.ndim, self.ndim)\) 范围。
ValueError - 参数 chunks 不是正数。

支持平台：

Ascend

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> input_x = Tensor(np.arange(9).astype("float32"))
>>> output = input_x.chunk(3, dim=0)
>>> 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]))

mindspore.Tensor.chunk(chunks, axis=0)

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

说明