Cupy sum 方法给出了奇怪的值

Posted 2023-02-16

【中文标题】Cupy sum 方法给出了奇怪的值

【英文标题】：cupy sum method gives strange value

【发布时间】：2021-11-07 10:33:45

【问题描述】：

我运行以下代码

import cupy as cp
x = cp.array([2.0, 3.0])
print(cp.sum(x))

然后它说 0.0 作为输出。此外，

x = cp.array([1, 2, 3, 4, 5, 6])
print(cp.sum(x))

给出 72340172838076673 作为输出。什么问题？以及如何正确使用 sum 方法？

【问题讨论】：

这很奇怪。您可以运行另一个 CUDA 库而没有任何问题吗？请同时粘贴cupy.show_config() 输出。

【参考方案1】：

cupy.sum(a, axis=None, dtype=None, out=None, keepdims=False)

沿给定轴返回数组的总和。 并且默认轴是None，这意味着如果应用不指定轴，它将返回0。