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