Fix torchscript issue in ConvTranspose2d (#1917)

pytorch · Feb 25, 2020 · 2937d77 · 2937d77
1 parent 2160353
commit 2937d77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/ops/misc.py b/torchvision/ops/misc.py
@@ -42,7 +42,7 @@ def forward(self, x):
                 list(self.output_padding),
             )
         ]
-        output_shape = [x.shape[0], self.bias.shape[0]] + output_shape
+        output_shape = [x.shape[0], self.out_channels] + output_shape
         return _new_empty_tensor(x, output_shape)
 
     def super_forward(self, input, output_size=None):