UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.


內部實則是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]


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM