会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
吴莫愁258
博客园
首页
新随笔
联系
订阅
管理
2021年8月12日
expected Long (got Float)
摘要: 不同类型的Tensor之间不能互转 torch.LongTensor(labels) 解决方法:先转化成numpy labels=torch.LongTensor(labels.numpy())
阅读全文
posted @ 2021-08-12 20:32 吴莫愁258
阅读(434)
评论(0)
推荐(0)
编辑
降维和升维
摘要: unsqueeze(arg)是增添第arg个维度为1,以插入的形式填充 squeeze(arg)是删除第arg个维度(如果当前维度不为1,则不会进行删除)
阅读全文
posted @ 2021-08-12 20:27 吴莫愁258
阅读(90)
评论(0)
推荐(0)
编辑
公告