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 问题跟踪器上。答案是：

没有，抱歉。而且恐怕我们不太可能添加这样的东西[.]