↓ 44 callersFunctionpermute_to_N_HWA_K Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
cvpods/modeling/meta_arch/retinanet.py:18
↓ 35 callersMethodfrom_tensors Args: tensors: a tuple or list of `torch.Tensors`, each of shape (Hi, Wi) or (C_1, ..., C_K, Hi, Wi) where K >= 1
cvpods/structures/image_list.py:50
↓ 10 callersFunctionpermute_to_N_HWA_K Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
playground/detection/coco/free_anchor/free_anchor.res50.fpn.coco.800size.1x/retinanet.py:18