loss = F.cross_entropy(output, labels)

output:网络的全连接层的输出,值可能是有正有负的.例如[1.56,-2.43]等.
labels:正常标签,例如一共5个类别,值就是0-4

更多推荐