会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
consolexinhun
新随笔
管理
2021年1月20日
pytorch one hot
摘要: print(torch.nn.functional.one_hot(t, num_classes=7)) 有个坑,使用的时候必须转换为 torch.int64 类型,不然会报错 t = t.to(torch.int64)
阅读全文
posted @ 2021-01-20 16:48 consolexinhun
阅读(310)
评论(0)
推荐(0)
编辑