您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python:将numpy默认为float32

Python:将numpy默认为float32

不是我知道的。您要么在调用任何数组的构造函数时需要显式指定dtype,要么将数组传递给float32(使用ndarray.astype方法)之前将其强制转换为float32(我认为这是问题所在)? )。如果您真正担心的是GPU情况,那么我赞成后者-在没有对numpy广播规则和非常精心设计的代码非常透彻理解的情况下,尝试使所有内容保持单一精度可能会变得非常烦人。

另一种选择是创建自己的方法,这些方法会使标准numpy构造函数(例如numpy.zeros,numpy.ones,numpy.empty)过载。这应该非常接近于将所有内容保持在float32中。

python 2022/1/1 18:39:04 有257人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶