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
样例:
>>> import numpy as np >>> import mindspore >>> from mindspore import Tensor >>> 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)