运行pytorch出现了这个错误:
Invalid index in scatter at /pytorch/aten/src/TH/generic/THTensorEvenMoreMath.cpp:151
这个问题是因为scatter造成的,scatter是pytorch自带的函数,这里有个严格的规定,就是index里面的最大值不能比维度高,不然就会下标越界。
scatter_(input, dim, index, src)或者input.scatter_(,dim, index, src)
这里我是使用的后者
举个例子就清楚了:
input.shape=(1,150,256,256)
index.shape=(1,1,256,256)
dim=1
src=1.0
因为input想做的事情是,将index里面的数值作为下标,而这个数值我们有可能就是图像,也就是0-255的数值,但是我们input.shape的第二个位置是150,很有可能我们图像的像素值有大于150的,这时就会出现下标越界,因为scatter_函数是将index里面的值作为下标。
因此要解决这个错误,当然是将像素值进行调整,调整到比150小就可以了