会员
周边
众包
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
星辰
博客园
首页
新随笔
联系
订阅
管理
2022年3月24日
pytorch CrossEntropyLoss() 默认转换one-hot编码
摘要: 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 一点飞鸿
阅读(441)
评论(0)
推荐(0)
编辑