Sometimes the output tensor from softmax contains NaN (not a number) ... ex = torch.exp(preds/temperature) return ex / torch.sum(ex, axis=0).
確定! 回上一頁