这个问题的错误提示很明显,索引必须是一个long、byte或bool tensor,但p_inds是一个floatTensor类型的tensor,因此需要将其转化为long类型。 下面是可能的解决方案:
检查代码中是否有将tensor类型转化为float类型的操作,如果有,可以考虑将其改为long类型或bool类型。
在p_inds被使用之前,可以使用.long()将其转化为long类型。代码可以像下面一样:
p_inds, n_inds = p_inds.long(), n_inds.long()
p_local_features = local_features[p_inds] # 使用p_inds进行索引,p_inds是long类型的tensor