pymc.math.sum

pymc.math.sum(input, axis=None, dtype=None, keepdims=False, acc_dtype=None)

计算张量 input 沿指定轴(轴)的总和。

当轴为 None（默认值）时，求和操作在展平的张量上进行。

如需完整文档，请参阅 Sum。特别注意在使用自定义 acc_dtype 时的重要警告。

参数:

keepdims: bool

如果设置为 True，被缩减的轴将作为尺寸为1的维度保留在结果中。使用此选项，结果将与原始张量正确广播。