若是直接使用argmax() 輸入標籤結果,那就有可能會產生這個報錯。 範例. 比如說我們以以下這樣的資料輸入CrossEntropyLoss() 當中:. import torch import ...
確定! 回上一頁