pytorch的nn.CrossEntropyLoss()函数使用方法

nn.CrossEntropyLoss()函数计算交叉熵损失

用法:

# output是网络的输出,size=[batch_size, class]
#如网络的batch size为128,数据分为10类,则size=[128, 10]
 
# target是数据的真实标签,是标量,size=[batch_size]
#如网络的batch size为128,则size=[128]
 
crossentropyloss=nn.CrossEntropyLoss()
crossentropyloss_output=crossentropyloss(output,target)

pytorch的nn.CrossEntropyLoss()函数使用方法_第1张图片

你可能感兴趣的:(DL)