提交 d15bdccf authored 作者: abergeron's avatar abergeron

Merge pull request #3445 from nouiz/work_around_dnn_pool_infer_shape

Work around dnn pool infer shape
......@@ -1512,10 +1512,15 @@ Linear Algebra
.. function:: dot(X, Y)
For 2-D arrays it is equivalent to matrix multiplication, and for
1-D arrays to inner product of vectors (without complex
conjugation). For N dimensions it is a sum product over the last
axis of a and the second-to-last of b:
:param X: left term
:param Y: right term
:type X: symbolic matrix or vector
:type Y: symbolic matrix or vector
:type X: symbolic tensor
:type Y: symbolic tensor
:rtype: symbolic matrix or vector
:return: the inner product of `X` and `Y`.
......
......@@ -1457,6 +1457,8 @@ class GpuDnnPool(DnnBase):
return Apply(self, [img, desc], [img.type()])
def infer_shape(self, node, shape):
if not node.inputs[1].owner:
raise theano.tensor.ShapeError()
desc = node.inputs[1].owner.op
nd = desc.get_ndim()
w = desc.ws
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论