基类:实现tensor的转换,主要将4D tensor转换成最后预测格式(N, H*W*A, K) ... pred_boxes2, pred_boxes3, pred_boxes4), dim=2).flatten(1)
確定! 回上一頁