不過在PyTorch 裡cross-entropy 因為input 是output layer 的值而不是softmax 後的probability,所以其實內部也在做log-softmax + nll,也不用先softmax。 那取log-softmax ...
確定! 回上一頁