提交 9781c838 authored 作者: Frederic Bastien's avatar Frederic Bastien

Update Python shape to be the same as the GPU. Cosmetic change only.

上级 ecc1b377
...@@ -220,7 +220,7 @@ class Pool(Op): ...@@ -220,7 +220,7 @@ class Pool(Op):
elif st[0] >= ds[0]: elif st[0] >= ds[0]:
nr = (r - 1) // st[0] + 1 nr = (r - 1) // st[0] + 1
else: else:
nr = max(0, (r - 1 - ds[0]) // st[0] + 1) + 1 nr = max(0, (r - 1 - ds[0] + st[0]) // st[0]) + 1
if isinstance(c, theano.Variable): if isinstance(c, theano.Variable):
nc = tensor.switch(tensor.ge(st[1], ds[1]), nc = tensor.switch(tensor.ge(st[1], ds[1]),
...@@ -230,7 +230,7 @@ class Pool(Op): ...@@ -230,7 +230,7 @@ class Pool(Op):
elif st[1] >= ds[1]: elif st[1] >= ds[1]:
nc = (c - 1) // st[1] + 1 nc = (c - 1) // st[1] + 1
else: else:
nc = max(0, (c - 1 - ds[1]) // st[1] + 1) + 1 nc = max(0, (c - 1 - ds[1] + st[1]) // st[1]) + 1
rval = list(imgshape[:-2]) + [nr, nc] rval = list(imgshape[:-2]) + [nr, nc]
return rval return rval
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论