提交 1e0b7bcd authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

flake8

上级 11c3bfa8
...@@ -2300,7 +2300,7 @@ class AbstractConv_gradInputs(BaseAbstractConv): ...@@ -2300,7 +2300,7 @@ class AbstractConv_gradInputs(BaseAbstractConv):
mshp0 = mat.shape[0] // self.num_groups mshp0 = mat.shape[0] // self.num_groups
mshp1 = mat.shape[1] * self.num_groups mshp1 = mat.shape[1] * self.num_groups
mat = mat.reshape((self.num_groups, mshp0) + mat.shape[1:]) mat = mat.reshape((self.num_groups, mshp0) + mat.shape[1:])
mat = mat.transpose((1, 0, 2) + tuple(range(3, 3+self.convdim))) mat = mat.transpose((1, 0, 2) + tuple(range(3, 3 + self.convdim)))
mat = mat.reshape((mshp0, mshp1) + mat.shape[-self.convdim:]) mat = mat.reshape((mshp0, mshp1) + mat.shape[-self.convdim:])
return mat return mat
kern = correct_for_groups(kern) kern = correct_for_groups(kern)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论