提交 d73bd46d authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix broadcastable pattern in DimShuffle, in sparse max-pooling.

上级 6c74e56c
......@@ -638,6 +638,6 @@ def max_pool(images, imgshp, maxpoolshp):
tensor.as_tensor(imgshp[0]))
out2 = tensor.reshape(out1, pshape, ndim=3);
out3 = tensor.DimShuffle((False,)*3, (0,2,1))(out2)
out3 = tensor.DimShuffle(out2.broadcastable, (0,2,1))(out2)
return tensor.flatten(out3,2), outshp
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论