nn.functional.softmax()输出形状

nn.functional.softmax()函数的输出形状与输入形状相同，即如果输入是形状为（batch_size，num_classes）的张量，则输出也是形状为（batch_size，num_classes）的张量。softmax函数会对每个样本的所有类别的分数进行归一化，使得每个类别的分数都在0到1之间，并且所有类别的分数之和为1。因此，输出张量的每一行都是一个概率分布，表示对应样本属于每个类别的概率。