nn.softmax

nn.functional.softmax是PyTorch中的一个函数，用于计算softmax函数。softmax函数是一种常用的激活函数，可以将一个向量转化为概率分布，常用于多分类问题中。该函数的输入是一个张量，输出是一个与输入张量形状相同的张量，其中每个元素都是输入张量对应位置的指数函数值除以所有元素的指数函数值之和。