Pytorch报错:RuntimeError : bool value of Tensor with more than one value is ambiguous. 的可能原因

1、使用损失函数的时候,不能直接写loss = nn.CrossEntropyLoss(input, target)
或者是nn.NLLLoss (input, target),必须先把求损失的库函数定义到一个自己命名的函数再调用。
例如:
loss_fn = nn.NLLLoss()
loss = loss_fn(input, target)
2、定义库函数的时候要记得加括号:
是loss_fn = nn.NLLLoss()
而不是loss_fn = nn.NLLLoss

你可能感兴趣的:(pytorch,pytorch)