ce_loss(X * 1000, torch.argmax(X,dim=1)) # tensor(0.) nn.CrossEntropyLoss 與logits一起使用,以利用log sum技巧 ...
確定! 回上一頁