【解决办法】torch交叉熵使用时遇到 Dimension out of range

简述

其实这个问题我很久以前用pytorch写程序的时候就遇到过这个问题，当时纠结了很久之后最后解决了。当时本来就想来写个东西来记录下避免其他人也遇到这样的问题。但后面我个菜鸡就完全忘记了emmmm

展示说明

不同于我以往直接给出结论。这次我会通过展示的方式较为详细的讲解下。

• 导入包

import torch
import torch.nn as nn

• 创造这个损失函数类

loss  = nn.CrossEntropyLoss()

• 随机生成下这个数据A

A = torch.randn(3, 5, requires_grad=True)

内容如下：

A
tensor([[