未验证 提交 056b8741 编写于 作者: L lzzyzlbb 提交者: GitHub

fix the order of unfold parameters (#34156)

* fix the order of unfold parameters
上级 6fbb975d
...@@ -1428,8 +1428,13 @@ class Unfold(layers.Layer): ...@@ -1428,8 +1428,13 @@ class Unfold(layers.Layer):
self.name = name self.name = name
def forward(self, input): def forward(self, input):
return F.unfold(input, self.kernel_sizes, self.dilations, self.paddings, return F.unfold(
self.strides, self.name) input,
kernel_sizes=self.kernel_sizes,
strides=self.strides,
paddings=self.paddings,
dilations=self.dilations,
name=self.name)
def extra_repr(self): def extra_repr(self):
name_str = ', name={}'.format(self.name) if self.name else '' name_str = ', name={}'.format(self.name) if self.name else ''
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册