Python：将numpy默认为float32

Question

是否有任何干净的方法来设置numpy以使用float32值而不是全局使用float64？

Answer 1

不是我知道的。 您需要在调用任何数组的构造函数时显式指定dtype，或者在将数组传递给GPU代码之前将数组转换为float32（使用ndarray.astype方法）（我认为这是问题所针对的？ ）。 如果是你真正担心的GPU情况，我赞成后者 - 如果没有对numpy广播规则和非常精心设计的代码进行非常透彻的理解，尝试保持所有内容都会变得非常烦人。

另一种方法可能是创建自己的方法，重载标准的numpy构造函数（如numpy.zeros，numpy.ones，numpy.empty）。 这应该非常接近将所有内容保存在float32中。

Answer 2

这个问题出现在NumPy问题跟踪器上。 答案是：

对不起，没有。 而且我担心我们不太可能添加这样的东西[。]

Answer 3

您可以通过以下方式重载每个功能：

def array(*args, **kwargs):
    kwargs.setdefault("dtype", np.float32)
    return np.array(*args, **kwargs)

由njsmith在github上发布