提交 bf4123cc authored 作者: Frederic Bastien's avatar Frederic Bastien

fix convolve2() when dx!=1 or dy!=1.

上级 ef510880
...@@ -254,7 +254,8 @@ def convolve2(kerns, kshp, nkern, images, imshp, bsize, step=(1,1), ...@@ -254,7 +254,8 @@ def convolve2(kerns, kshp, nkern, images, imshp, bsize, step=(1,1),
kernrshp = tensor.as_tensor([nkern, nvis_dim] + list(kshp)) kernrshp = tensor.as_tensor([nkern, nvis_dim] + list(kshp))
kerntensor = tensor.reshape(kerns, kernrshp) kerntensor = tensor.reshape(kerns, kernrshp)
convop = ConvOp(imshp, kshp, nkern, bsize, 1, 1, output_mode=mode,**d) convop = ConvOp(imshp, kshp, nkern, bsize, step[0], step[1],
output_mode=mode, **d)
convout = convop(imtensor, kerntensor) convout = convop(imtensor, kerntensor)
if bias: if bias:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论