pytorch报错: scatter_cpu_(): Expected self.dtype to be equal to src.dtype

最近在初学pytorch,然后在运行代码的时候出现了这个错误：

import torch as t
a = t.arange(0,16).view(4,4)
index = t.LongTensor([[0,1,2,3],[3,2,1,0]]).t()
b = a.gather(1,index)
c = t.zeros(4,4)
c.scatter_(1,index,b)
print(c)
----------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-4-b907583c1f90> in <module>4 b = a.gather(1,index)5 c = t.zeros(4,4)
----> 6 c.scatter_(1,index,b)7 print(c)RuntimeError: scatter_cpu_(): Expected self.dtype to be equal to src.dtype

后来去了下pytorch forums（那里听说氛围不错，所以我就抱着尝试的心态发了个求助帖），经过指点发现，原来这个错误是指的scatter_()函数需要实参的dtype一致，而t.arange()方法生成的tensor是torch.int64的，t.LongTensor(),t.randn(),t.zeros()生成的tensor是torch.float32的。所以只需要在这一句末尾加上.float()就可以正常运行了。

pytorch报错: scatter_cpu_(): Expected self.dtype to be equal to src.dtype相关推荐

安装pytorch报错及解决
环境,win10,64位,Anaconda4.10.3,Cuda 9.2.148,cudnn应该为v7.4.2.24 其中在Anaconda中好像安装tensorflow-gpu 1.9 ,pytho ...
pycharm安装pytorch报错提示系列问题 torch 包找不到因为pip版本低，结果升级了pip从19.3到20.2 4又提示高版不支持torch安装
pycharm安装pytorch报错提示系列问题 torch 包找不到因为pip版本低,结果升级了pip从19.3到20.2 4又提示高版不支持torch安装 DEPRECATION: The - ...
pytorch报错Unable to get repr for
pytorch报错Unable to get repr for 标签出现了-1,2或者其他值,loss计算之后, 报错:CUDA error: device-side assert triggered ...
react native报错：Expected a component class,got[object object]
正在写RN的视图部分,突然模拟器报错:Expected a component class,got[object object].找也找不到原因.后来才发现是自己粗心大意造成的. 错误代码: < ...
解决Vue项目报错：Expected indentation of 2 spaces but found 4. eslint(indent) [8, 1]的方法
解决Vue项目报错:Expected indentation of 2 spaces but found 4. eslint(indent) [8, 1]的方法报错如下问题分析明显从图中可以以看 ...
Pytorch | 报错The given NumPy array is not writeable,and PyTorch does not support non-writeable tensor
Pytorch | 报错 The given NumPy array is not writeable,and PyTorch does not support non-writeable tenso ...
读取导入csv csv报错iterable expected, not float
示例代码 import pandas as pd import re import csv data = pd.read_csv('nuojia.csv', encoding='utf-8') # p ...
Python报错“IndentationError: expected an indented block“
问题描述: 一段python代码中使用了if name == 'main':,运行程序是报错"IndentationError: expected an indented block&quo ...
DDP报错::RuntimeError: Expected all tensors to be on the same device, but found at least two devic
DDP报错 RuntimeError: Expected all tensors to be on the same device, but found at least two devices, c ...

pytorch报错: scatter_cpu_(): Expected self.dtype to be equal to src.dtype

pytorch报错: scatter_cpu_(): Expected self.dtype to be equal to src.dtype相关推荐

最新文章

热门文章