IndexError: too many indices for tensor of dimension 2

bug场景

打印的报错如下

代码中self.geo_mask是维度为20 * 20的tensor

sh_mx是维度为20 * 20的<class 'numpy.ndarray'>，

self.far_mask_delta为3,int类型

明明维度sh_mx >= 0和self.geo_mask的维度是匹配的，但总是报这个IndexError的问题。

解决

将sh_mx >= 0转化为tensor，思路来自于下面的博客。

如果我们遇到了“IndexError: too many indices for tensor of dimension 0”错误，可通过查看数据集的标签维度来确定问题所在。例如，使用print(target.shape)语句来打印标签的维度。

一种常见的错误是标签数组的维度仅为零。这通常是由于数据集的标签未正确转换为Tensor格式导致的。为了解决这个问题，我们需要在转换数据集时确保标签的正确维度，通常是一维数组。
Pytorch 错误：“IndexError: too many indices for tensor of dimension 0”|极客笔记 (deepinout.com)https://deepinout.com/pytorch/pytorch-questions/130_pytorch_pytorch_dataloader_indexerror_too_many_indices_for_tensor_of_dimension_0.html