不是我知道的。您要么在调用任何数组的构造函数时需要显式指定dtype,要么将数组传递给float32(使用ndarray.astype方法)之前将其强制转换为float32(我认为这是问题所在)?
)。如果您真正担心的是GPU情况,那么我赞成后者-在没有对numpy广播规则和非常精心设计的代码非常透彻理解的情况下,尝试使所有内容保持单一精度可能会变得非常烦人。
另一种选择是创建自己的方法,这些方法会使标准numpy构造函数(例如numpy.zeros,numpy.ones,numpy.empty)过载。这应该非常接近于将所有内容保持在float32中。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)