![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
飞鱼~
一个有理想的码农小白
展开
-
Pyotrch常见bug:Found dtype Long but expected
在loss.backward()的时候出现Found dtype Long but expected。但是看了一下loss的类型,发现是flaot32没问题。看了各种解决方案,发现一般是loss函数的输入类型不一致会导致这个问题。比如我用的是nn.MSELoss(),两个输入必须都转换为float才行。原创 2022-12-26 22:12:59 · 882 阅读 · 0 评论 -
pytorch中使用tensor作为另一个tensor的索引 (torch.gather)
使用tensor对tensor进行索引的方法原创 2022-10-20 18:30:57 · 1509 阅读 · 1 评论