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
·
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
12
0- 0
已为社区贡献3条内容
所有评论(0)