2021
07-09
07-09
解决numpy和torch数据类型转化的问题
在实际计算过程中,float类型使用最多,因此这里重点介绍numpy和torch数据float类型转化遇到的问题,其他类型同理。numpy数据类型转化numpy使用astype转化数据类型,float默认转化为64位,可以使用np.float32指定为32位#numpy转化float类型a=np.array([1,2,3])a=a.astype(np.float)print(a)print(a.dtype)[1.2.3.]float64不要使用a.dtype指定数据类型,会使数据丢失#numpy转化float类型b=np.array([1,2,3])b.dtype=np.float...
继续阅读 >