RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #2 ‘mat1‘

原创于 2020-10-01 00:26:44 发布 · 287 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#numpy #pytorch

问题记录专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一个常见的数据类型问题，即在使用numpy数组时遇到的RuntimeError: expected object of scalar type Float but got scalar type Double。文章提供了简单的解决方案：在初始化DataSet时，明确指定dtype为np.float32。

解决方法：

在DataSet初始化函数中，显示指明dtype为np.float32
self.datas = np.array(datas ,dtype=np.float32)

参考链接：

https://stackoverflow.com/questions/60020184/runtimeerror-expected-object-of-scalar-type-float-but-got-scalar-type-double-fo