摘要: import torchpredict = torch.randn((4,3))# crossentropyloss不需要predict的概率为1,predict为logits# predict = torch.nn.functional.softmax(predict,dim = 1)target 阅读全文
posted @ 2022-03-24 22:46 一点飞鸿 阅读(414) 评论(0) 推荐(0) 编辑