学习《深度学习框架PyTorch:入门与实践》,遇到报错
RuntimeError: scatter_cpu_(): Expected self.dtype to be equal to src.dtype
原因
我们使用scatter_ 将数据放回去时,类型不一致
解决方法
将类型转换为一致的,即可以加入
b = b.float()
如下图所示:
学习《深度学习框架PyTorch:入门与实践》,遇到报错
RuntimeError: scatter_cpu_(): Expected self.dtype to be equal to src.dtype
原因
我们使用scatter_ 将数据放回去时,类型不一致
解决方法
将类型转换为一致的,即可以加入
b = b.float()
如下图所示: