如何将float类型tensor转为double
收藏
如何将float64类型的tensor转换为double类型的?有什么API可以做到吗?
0
收藏
请登录后评论
我都是通过numpy转的
一般tensor在模型里用的时候不能变精度吧
paddle一般都要求类型为float32,标签有时要求int64
我一般都是np一大串处理后面加个.astype('float32'),这样就转成float32类型了
问题解决了。这里纠正一下自己的问题,在paddle中float64就是double类型的。float32是float类型的。
我的遇到的问题是,在计算两个矩阵相乘的时候报错,是因为计算的时候一个是float类型一个是double类型,然后提示我转为double类型。
最后就是paddle类型是可以直接转类型的,直接用paddleTensor.astype('float32') 就可以将数据转换为float类型。
下面是一个将float类型tensor转为double类型的例子:
numpy好用,可以互转。
原来paddle的tensor也加了这个方法
有时候,我觉着照着numpy或torch的api试试,paddle会有惊喜~~
另外paddle将各种elementwise的操作都用运算符重载了,这个也是超酷
我是误用才发现这个的