![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
陶云松
这个作者很懒,什么都没留下…
展开
-
RuntimeError: Expected object of scalar type Float but got scalar type Int for argument #2 'other'
不同类型的是不相等的,并且无法比较import torcha=torch.tensor([1],dtype=torch.float)b=torch.tensor([1],dtype=torch.int)#tensor 的1无论什么情况都与普通的1相等#不同类型的是不相等的,并且无法比较if a==b: print((a==b)) print((a==b).long(...原创 2019-07-14 20:41:06 · 4682 阅读 · 0 评论 -
pytorch KeyError: tensor(1)
pytorch中的tensor不能作为字典的键,输入下面代码会产生错误import torch'''pytorch中的tensor和list不能混用'''a=torch.tensor([1,2,4])b={1:'a',2:'b',4:'d'}c=[b[o] for o in a]print(c)KeyError: tensor(1)正确的是要在循环中临时转成nump...原创 2019-07-14 20:56:41 · 7848 阅读 · 3 评论 -
Pytorch tensor.view_as的用法
示例代码import torcha=torch.tensor([1,2,3])b=torch.tensor([2,2,4])print(a.detach().view_as(b))一般是用来做比较的时候使用,确保两个比较的数据,维度是一致的。...原创 2019-07-15 08:53:09 · 3523 阅读 · 0 评论 -
pytorch torch.uint8的用法及uint8求和sum()
uint8是8位的,表示十进制数0-255,如果大于255则进一位,例如256显示就变成0示例代码如下c=256a=torch.tensor(1,dtype=torch.uint8)tensor=[a for i in range(c)]print(sum(tensor))结果tensor(0, dtype=torch.uint8)改正方法对uint8求和要使用...原创 2019-07-15 09:38:41 · 13100 阅读 · 0 评论