↓ 1 callersFunctionpermute_all_cls_and_box_to_N_HWA_K_and_concat Rearrange the tensor layout from the network output, i.e.: list[Tensor]: #lvl tensors of shape (N, A x K, Hi, Wi) to per-image prediction
OpenSelfSup/benchmarks/detection/detection/projects/TensorMask/tensormask/arch.py:23