提交 9ef594de authored 作者: Sina Honari's avatar Sina Honari

correcting DownsampleFactorMaxGrad, issue #2196

上级 bf64aee7
...@@ -375,7 +375,7 @@ class DownsampleFactorMaxGrad(Op): ...@@ -375,7 +375,7 @@ class DownsampleFactorMaxGrad(Op):
for row_ind in xrange(row_st, row_end): for row_ind in xrange(row_st, row_end):
for col_ind in xrange(col_st, col_end): for col_ind in xrange(col_st, col_end):
if (maxout[n, k, r, c] == x[n, k, row_ind, col_ind]): if (maxout[n, k, r, c] == x[n, k, row_ind, col_ind]):
gx[n, k, row_ind, col_ind] = gz[n, k, r, c] gx[n, k, row_ind, col_ind] += gz[n, k, r, c]
gx_stg[0] = gx gx_stg[0] = gx
def infer_shape(self, node, in_shapes): def infer_shape(self, node, in_shapes):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论