Skip to content

Commit 96a281d

Browse files
ezyangsoumith
authored andcommitted
Add one more missing self.dilation parameter. (pytorch#1392)
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
1 parent 94b147f commit 96a281d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/modules/conv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -421,7 +421,7 @@ def forward(self, input, output_size=None):
421421
output_padding = self._output_padding(input, output_size)
422422
return F.conv_transpose1d(
423423
input, self.weight, self.bias, self.stride, self.padding,
424-
output_padding, self.groups)
424+
output_padding, self.groups, self.dilation)
425425

426426

427427
class ConvTranspose2d(_ConvTransposeMixin, _ConvNd):

0 commit comments

Comments
 (0)