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