Pytorch——代码导致的异常报错：RuntimeError: CUDA out of memory.

报错截图

解决方法：

如果你的程序之前是正常的，在经过修改后报该错误，说明是你的tensor在操作时，可能在维度不同的时候就进行了叠加，导致报错。
在该报错语句前检查检查你输入的tensor维度，一般问题都在这。