Invalid index in scatter at /pytorch/aten/src/TH/generic/THTensorEvenMoreMath.cpp:151

运行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小就可以了