mindspore.Tensor.sum_to_size

mindspore.Tensor.sum_to_size(*size)

将原Tensor按照指定 size 进行求和。size 必须可以扩展到Tensor的shape。

参数：

• size (Union[tuple(int), int]) - 期望输出Tensor的shape。

返回：

Tensor，根据 size 对原Tensor进行求和的结果。

异常：

• ValueError - size 不能扩展成原Tensor的大小。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.random.randn(3, 3, 3, 3, 3, 3), mindspore.float32)
>>> output = x.sum_to_size((1, 3, 1, 3))
>>> print(output.shape)
(1, 3, 1, 3)