Skip to content

P
pytensor
提交 e4089d8d authored 作者: Frederic's avatar Frederic
浏览文件
操作

[BUG] fix infer_shape of DownsampleFactorMaxGradGrad. This fix the test…

[BUG] fix infer_shape of DownsampleFactorMaxGradGrad. This fix the test theano/sandbox/cuda/tests/test_blas.py:test_downsample.
上级 6f86147d
显示空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
theano/tensor/signal/downsample.py
浏览文件 @ e4089d8d
...@@ -1053,7 +1053,7 @@ class DownsampleFactorMaxGradGrad(Op): ...@@ -1053,7 +1053,7 @@ class DownsampleFactorMaxGradGrad(Op):
ggz[n, k, r, c] = ggx_padded[n, k, row_ind, col_ind] ggz[n, k, r, c] = ggx_padded[n, k, row_ind, col_ind]
def infer_shape(self, node, in_shapes): def infer_shape(self, node, in_shapes):
return [in_shapes[0]] return [in_shapes[1]]
def c_code(self, node, name, inp, out, sub): def c_code(self, node, name, inp, out, sub):
if self.mode != 'max': if self.mode != 'max':
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论