提交 59e08c0d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix 3d convs with groups.

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