使用torch.softmax会将原本torch.Size([2, 4, 2])的张量变成torch.Size([8, 2])，
改成torch.nn.functional.softmax则不会改变张量维度，但是要记得指定softmax计算的维度torch.nn.functional.softmax（outputs, dim=-1）
版权声明：本文为weixin_44003026原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。
使用torch.softmax会将原本torch.Size([2, 4, 2])的张量变成torch.Size([8, 2])，
改成torch.nn.functional.softmax则不会改变张量维度，但是要记得指定softmax计算的维度torch.nn.functional.softmax（outputs, dim=-1）