问为什么我会出错：“一个以上值的张量的布尔值不明确”
EN

Stack Overflow用户

提问于 2022-06-21 19:33:15

回答 1查看 492关注 0票数 1

我试图运行以下代码：

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

为了运行代码，我需要更改什么？

回答已采纳

发布于 2022-06-21 19:38:35

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

因此，您应该事先取消它：

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

>>> F.cross_entropy(r, l)

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/72706218

复制

相似问题

问为什么我会出错：“一个以上值的张量的布尔值不明确”EN