Код: Выделить всё
import torch
loss = nn.CrossEntropyLoss()
loss(torch.tensor((.1, .2)), torch.tensor((.3, .4)))
Код: Выделить всё
tensor(0.4811)Код: Выделить всё
nn.CrossEntropyLoss(torch.tensor((.1, .2)), torch.tensor((.3, .4)))Похоже, что loss(input, target) эквивалентно
Код: Выделить всё
torch.nn.functional.cross_entropy(input target)Подробнее здесь: https://stackoverflow.com/questions/798 ... ntropyloss
Мобильная версия