4、nn.CrossEntropyLoss. 需要注意的是,target输入必须是tensor long 类型(int64位) import torch # cross entropy loss pred = ...
確定! 回上一頁