Fix torchscript issue in ConvTranspose2d (#1917) (#2432)

Summary: Pull Request resolved: #2432 Reviewed By: zhangguanheng66 Differential Revision: D22437817 Pulled By: fmassa fbshipit-source-id: 4358dd941b90eb222d4ad7e211793bf21653be87
pytorch · Jul 9, 2020 · ac7604b · ac7604b
1 parent 72697fd
commit ac7604b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/ops/misc.py b/torchvision/ops/misc.py
@@ -43,7 +43,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):