提交 19b6c0a6 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Merge pull request #3987 from SinaHonari/issue3681

fixing make_node param passing for conv_grad (conv_transpose)
...@@ -222,7 +222,7 @@ def conv2d_grad_wrt_inputs(output_grad, ...@@ -222,7 +222,7 @@ def conv2d_grad_wrt_inputs(output_grad,
subsample=subsample, subsample=subsample,
filter_flip=filter_flip) filter_flip=filter_flip)
return grad_input_op(filters, input, output_grad_shape) return grad_input_op(filters, output_grad, input_shape)
def conv2d_grad_wrt_weights(input, def conv2d_grad_wrt_weights(input,
...@@ -326,7 +326,7 @@ def conv2d_grad_wrt_weights(input, ...@@ -326,7 +326,7 @@ def conv2d_grad_wrt_weights(input,
subsample=subsample, subsample=subsample,
filter_flip=filter_flip) filter_flip=filter_flip)
return gradWeight_op(input, output_grad, input_shape) return gradWeight_op(input, output_grad, filter_shape)
class BaseAbstractConv2d(Op): class BaseAbstractConv2d(Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论