【问题标题】:cupy sum method gives strange valueCupy sum 方法给出了奇怪的值
【发布时间】: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() 输出。

标签: python cupy


【解决方案1】:
cupy.sum(a, axis=None, dtype=None, out=None, keepdims=False)

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

【讨论】:

    猜你喜欢
    • 2018-11-21
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2017-11-29
    • 2013-07-25
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多