expected scalar type Double but found Float

@expected scalar type Double but found Float

虽然报错需要double，但使用.astype(‘double’)还是会报错。
把.astype(‘double’)改成.astype(‘float32’)就可以了。

问题解决，原因不明，有会的大神还请赐教。

参照
[1]:https://blog.youkuaiyun.com/ttdxtt/article/details/112347283