会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
抚琴尘世客
但行好事 莫问前程
博客园
首页
新随笔
联系
订阅
管理
2023年4月14日
Pytorch one-hot编码
摘要: 1. 引言 在我们做分割任务时,通常会给一个mask,但训练时要进行onehot编码。 2. code import torch if __name__ == '__main__': label = torch.zeros(size=(1, 4, 4), dtype=torch.int) label
阅读全文
posted @ 2023-04-14 10:03 抚琴尘世客
阅读(156)
评论(0)
推荐(0)
编辑
公告