[fbsync] Avoid creating a tensor of shape when not tracing (#7942)

Summary: Signed-off-by: Edward Z. Yang <[email protected]>

Reviewed By: matteobettini

Differential Revision: D49600781

fbshipit-source-id: 374a98652b1c9e57429edfbafb42caddf4f4365d

Author: NicolasHug

torchvision/models/detection/transform.py

@@ -31,8 +31,10 @@ def _resize_image_and_masks(
 ) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
     if torchvision._is_tracing():
         im_shape = _get_shape_onnx(image)
-    else:
+    elif torch.jit.is_scripting():
         im_shape = torch.tensor(image.shape[-2:])
+    else:
+        im_shape = image.shape[-2:]
 
     size: Optional[List[int]] = None
     scale_factor: Optional[float] = None