具有多个值的张量的布尔值在 Pytorch 中不明确

Posted 2023-03-29

【中文标题】具有多个值的张量的布尔值在 Pytorch 中不明确

【英文标题】：Bool value of Tensor with more than one value is ambiguous in Pytorch

【发布时间】：2019-03-27 13:35:00

【问题描述】：

我想在 pytorch 中创建一个模型，但我做不到 计算损失。 它总是返回 Tensor 的布尔值与更多 一个值不明确 实际上，我运行示例代码，它可以工作。

loss = CrossEntropyLoss()
input = torch.randn(8, 5)
input
target = torch.empty(8,dtype=torch.long).random_(5)
target
output = loss(input, target)

这是我的代码，

################################################################################
##
##
import torch
from torch.nn import Conv2d, MaxPool2d, Linear, CrossEntropyLoss, MultiLabelSoftMarginLoss
from torch.nn.functional import relu, conv2d, max_pool2d, linear, softmax
from torch.optim import adadelta
##
##
##  Train
Train = 
Train["Image"]    = torch.rand(2000, 3, 76, 76)
Train["Variable"] = torch.rand(2000, 6)
Train["Label"] = torch.empty(2000, dtype=torch.long).random_(2)
##
##
##  Valid
Valid = 
Valid["Image"]    = torch.rand(150, 3, 76, 76)
Valid["Variable"] = torch.rand(150, 6)
Valid["Label"]    = torch.empty(150, dtype=torch.long).random_(2)
################################################################################
##
##
##  Model
ImageTerm    = Train["Image"]
VariableTerm = Train["Variable"]
Pip = Conv2d(in_channels=3, out_channels=32, kernel_size=(3,3), stride=1, padding=0)(ImageTerm)
Pip = MaxPool2d(kernel_size=(2,2), stride=None, padding=0)(Pip)
Pip = Conv2d(in_channels=32, out_channels=64, kernel_size=(3,3), stride=1, padding=0)(Pip)
Pip = MaxPool2d(kernel_size=(2,2), stride=None, padding=0)(Pip)
Pip = Pip.view(2000, -1)
Pip = torch.cat([Pip, VariableTerm], 1)
Pip = Linear(in_features=18502, out_features=1000 , bias=True)(Pip)
Pip = Linear(in_features=1000, out_features=2 , bias=True)(Pip)
##
##
##  Loss
Loss = CrossEntropyLoss(Pip, Train["Label"])

错误在于Loss = CrossEntropyLoss(Pip, Train["Label"])， 谢谢。

【问题讨论】：

相关帖子：-***.com/questions/52946920/…-discuss.pytorch.org/t/…-discuss.pytorch.org/t/…-discuss.pytorch.org/t/…

【参考方案1】：

在您的最小示例中，您创建了“CrossEntropyLoss”类的对象“损失”。该对象能够将您的损失计算为

loss(input, target)

但是，在您的实际代码中，您尝试创建对象“Loss”，同时将 Pip 和标签传递给“CrossEntropyLoss”类构造函数。 相反，请尝试以下方法：

loss = CrossEntropyLoss()
loss(Pip, Train["Label"])

编辑（错误消息的解释）：当您尝试将张量转换为布尔值时，会出现错误消息Bool value of Tensor with more than one value is ambiguous。这种情况最常发生在将张量传递给 if 条件时，例如

input = torch.randn(8, 5)
if input:
    some_code()

CrossEntropyLoss 类构造函数的第二个参数需要一个布尔值。因此，在行中

Loss = CrossEntropyLoss(Pip, Train["Label"])

构造函数有时会尝试使用传递的张量 Train["Label"] 作为布尔值，这会引发上述错误消息。

【讨论】：

如果您在某处引用该错误的实际含义会很好

【参考方案2】：

您不能直接使用CrossEntropyLoss 类。你应该在使用它之前实例化这个类。

原代码：

loss = CrossEntropyLoss(Pip, Train["Label"])

应替换为：

loss = CrossEntropyLoss()
loss(Pip, Train["Label"])

【参考方案3】：

首先实例化损失

L = CrossEntropyLoss()

然后计算损失

L(y_pred, y_true)

这将修复错误。