box_target = torch.where(
cls_target[..., None] == -1, box_target.new_tensor(0), box_target
)
cls_target[b,23] cls_target[..., None] [b,23,1]
box_target[b,23,10]
torch.where
好记性不如烂键盘---点滴、积累、进步！
深入了解每一个知识点
box_target = torch.where(
cls_target[..., None] == -1, box_target.new_tensor(0), box_target
)
cls_target[b,23] cls_target[..., None] [b,23,1]
box_target[b,23,10]