未验证 提交 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):
self.name = name
def forward(self, input):
return F.unfold(input, self.kernel_sizes, self.dilations, self.paddings,
self.strides, self.name)
return F.unfold(
input,
kernel_sizes=self.kernel_sizes,
strides=self.strides,
paddings=self.paddings,
dilations=self.dilations,
name=self.name)
def extra_repr(self):
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.
先完成此消息的编辑!
想要评论请 注册