sponge.function.keepdims_sum
- sponge.function.keepdims_sum(x: Tensor, axis: Union[int, Tuple[int], List[int]] = ())[源代码]
对轴中从维度为 x 减小到维度为 1 的元素求和，并且输出和输入的维度相同。
- 参数：
x (Tensor[Number]) - 输入张量。要约简的张量的 dtype 是 number。 \((N,*)\) 其中的 \(*\) 代表任意数量的附加维度，其维度应小于 8。
axis (Union[int, tuple(int), list(int)]) - 要减小的维度。默认值：()，减小所有维度。 只允许常量值。必须在范围 [-rank(x), rank(x))。
- 输出：
Tensor。和 x 有相同的dtype。
- 支持平台：
AscendGPUCPU