2023年3月10日
pytorch tensor cat for循环每次的tensor
摘要： detections_out = torch.tensor([], dtype=torch.float32).to(model_out[0].device) for i in range(5): detections = torch.cat([bboxes, scores, clses], dim= 阅读全文
posted @ 2023-03-10 13:56 无左无右 阅读(109) 评论(0) 推荐(0) 编辑