[pytorch]关于cross_entropy函数
·
loss = F.cross_entropy(output, labels)
output:网络的全连接层的输出,值可能是有正有负的.例如[1.56,-2.43]等.
labels:正常标签,例如一共5个类别,值就是0-4
更多推荐
loss = F.cross_entropy(output, labels)
output:网络的全连接层的输出,值可能是有正有负的.例如[1.56,-2.43]等.
labels:正常标签,例如一共5个类别,值就是0-4
更多推荐
所有评论(0)