您的当前位置:首页pytorch中CrossEntropyLoss的使用

pytorch中CrossEntropyLoss的使用

来源:乌哈旅游
pytorch中CrossEntropyLoss的使⽤

CrossEntropyLoss 等价于 softmax+log+NLLLossLogSoftmax等价于softmax+log

# ⾸先定义该类

loss = torch.nn.CrossEntropyLoss()#然后传参进去loss(input, target)

input维度为N*C,是⽹络⽣成的值,N为batch_size,C为类别数;target维度为N,是标注值,⾮one-hot类型的值;

因篇幅问题不能全部显示,请点此查看更多更全内容