Почему я не могу передать входные и целевые тензоры непосредственно в 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
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

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