pytorch int64的tensor怎么转换成float64

correct = correct.to(torch.float64)

https://www.jianshu.com/p/75dff8e7ed18

猜你喜欢

转载自blog.csdn.net/s1162276945/article/details/88662843