Почему я не могу передать входные и целевые тензоры непосредственно в nn.CrossEntropyLoss?Python

Программы на Python
Anonymous
Почему я не могу передать входные и целевые тензоры непосредственно в nn.CrossEntropyLoss?

Сообщение Anonymous »

на Python 3.13, torch 2.10.0+cu130

Код: Выделить всё

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)))
return RuntimeError: логическое значение Tensor с более чем одним значением неоднозначно?
Похоже, что loss(input, target) эквивалентно

Код: Выделить всё

torch.nn.functional.cross_entropy(input target)
, но я не могу найти его в документации CrossEntropyLoss. Откуда мне это знать?

Подробнее здесь: https://stackoverflow.com/questions/798 ... ntropyloss

Вернуться в «Python»