为什么我收到错误：“具有多个值的张量的布尔值不明确”

Question

我正在尝试运行以下代码：

l = torch.tensor([0, 1, 1, 1], requires_grad=False)
r = torch.rand(4, 2)

torch.nn.CrossEntropyLoss(r, l)

我收到错误：

RuntimeError: Boolean value of Tensor with more than one value is ambiguous

我在这里查看：Bool value of Tensor with more than one value is ambigacy in Pytorch但不明白答案。

我需要更改什么才能运行代码？

Answer 1

您正在操作的对象torch.nn.CrossEntropyLoss是 PyTorch 模块类，而不是函数。

因此，您应该提前初始化它：

>>> ce_loss = nn.CrossEntropyLoss()
>>> cel_loss(r, l)

或者使用函数式接口，即 torch.nn.functional.cross_entropy：

>>> F.cross_entropy(r, l)