[fbsync] Pass indexing param to meshgrid to avoid warning (#4645)

Reviewed By: datumbox

Differential Revision: D31898207

fbshipit-source-id: e23c0b73c6cad555f5247674c214e567eb87ba7f

Author: fmassa

test/test_io.py

@@ -23,7 +23,7 @@
 
 
 def _create_video_frames(num_frames, height, width):
-    y, x = torch.meshgrid(torch.linspace(-2, 2, height), torch.linspace(-2, 2, width))
+    y, x = torch.meshgrid(torch.linspace(-2, 2, height), torch.linspace(-2, 2, width), indexing="ij")
     data = []
     for i in range(num_frames):
         xc = float(i) / num_frames

torchvision/models/detection/anchor_utils.py

@@ -104,7 +104,7 @@ def grid_anchors(self, grid_sizes: List[List[int]], strides: List[List[Tensor]])
             # For output anchor, compute [x_center, y_center, x_center, y_center]
             shifts_x = torch.arange(0, grid_width, dtype=torch.int32, device=device) * stride_width
             shifts_y = torch.arange(0, grid_height, dtype=torch.int32, device=device) * stride_height
-            shift_y, shift_x = torch.meshgrid(shifts_y, shifts_x)
+            shift_y, shift_x = torch.meshgrid(shifts_y, shifts_x, indexing="ij")
             shift_x = shift_x.reshape(-1)
             shift_y = shift_y.reshape(-1)
             shifts = torch.stack((shift_x, shift_y, shift_x, shift_y), dim=1)
@@ -222,7 +222,7 @@ def _grid_default_boxes(
 
             shifts_x = ((torch.arange(0, f_k[1]) + 0.5) / x_f_k).to(dtype=dtype)
             shifts_y = ((torch.arange(0, f_k[0]) + 0.5) / y_f_k).to(dtype=dtype)
-            shift_y, shift_x = torch.meshgrid(shifts_y, shifts_x)
+            shift_y, shift_x = torch.meshgrid(shifts_y, shifts_x, indexing="ij")
             shift_x = shift_x.reshape(-1)
             shift_y = shift_y.reshape(-1)