內部實則是bool。
原:
idxs = torch.Tensor(idxs_0.shape[0]).byte()
idxs[:] = 1
idxs[idxs_0 < 0] = 0
idxs[idxs_end >= int(N / args.gps_step)] = 0
delta_p_gt = delta_p_gt[idxs]
idxs_0_bias = idxs_0[idxs]
idxs_end_bis = idxs_end[idxs]
可以看出,實則idxs內部為bool性質。
改后:
# 加一個.bool()
idxs = torch.Tensor(idxs_0.shape[0]).byte().bool()
idxs[:] = 1
idxs[idxs_0 < 0] = 0
idxs[idxs_end >= int(N / args.gps_step)] = 0
delta_p_gt = delta_p_gt[idxs]
idxs_0_bias = idxs_0[idxs]
idxs_end_bis = idxs_end[idxs]