在pytorch中float32为float类型,而float64则为double类型,注意tensor的数据类型。
解决方法,将数据进行转换,原因是我这里的dtype是float64,需要转成float 32
bbb=aa.view(40000,-1,2,2) bbb = bbb.to(torch.float32) bbb.dtype
转完就不会报错的

在pytorch中float32为float类型,而float64则为double类型,注意tensor的数据类型。
解决方法,将数据进行转换,原因是我这里的dtype是float64,需要转成float 32
bbb=aa.view(40000,-1,2,2) bbb = bbb.to(torch.float32) bbb.dtype
转完就不会报错的