cupy.sum#

cupy.sum(a, axis=None, dtype=None, out=None, keepdims=False)[source]#

返回数组沿给定轴的和。

参数:

a (cupy.ndarray) – 要求和的数组。
axis (int 或 int 序列) – 求和的轴。
dtype – 数据类型说明符。
out (cupy.ndarray) – 输出数组。
keepdims (bool) – 如果为 True，指定的轴将保留为长度为一的轴。

返回:

结果数组。

返回类型:

另请参阅