主要介绍了浅谈pytorch中torch.max和F.softmax函数的维度解释，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

在利用torch.max函数和F.Ssoftmax函数时，对应该设置什么维度，总是有点懵，遂总结一下： 首先看看二维tensor的函数的例子： import torch import torch.nn.functional as F input = torch.randn(3,4) print(input) tensor([[-0.5526, -0.0194, 2.1469, -0.2567], [-0.3337, -0.9229, 0.0376, -0.0801], [ 1.