Fix segfault in DeformConv2d when mask is None (#7632)

pytorch · May 26, 2023 · 81d12c3 · 81d12c3
1 parent 4a51822
commit 81d12c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/ops/deform_conv.py b/torchvision/ops/deform_conv.py
@@ -68,7 +68,7 @@ def deform_conv2d(
     use_mask = mask is not None
 
     if mask is None:
-        mask = torch.zeros((input.shape[0], 0), device=input.device, dtype=input.dtype)
+        mask = torch.zeros((input.shape[0], 1), device=input.device, dtype=input.dtype)
 
     if bias is None:
         bias = torch.zeros(out_channels, device=input.device, dtype=input.dtype)