pytorch CrossEntropyLoss测试

 

测试代码：

import torch

loss_fn = torch.nn.CrossEntropyLoss(reduce=True, weight=None)
input = torch.Tensor([0,0,0,0.5,0.6,.7])
target = torch.LongTensor(3).random_(5)
print(input)
print(target)
loss = loss_fn(input, target)

print(loss)