Hi. I’ve tried nn.CrossEntropyLoss as well with nn.CrossEntropyLoss(output, label) and nn.CrossEntropyLoss(output, label.flatten()) and both cases still return:
RuntimeError: bool value of Tensor with more than one value is ambiguous