提交 9481535c authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Import SpecifyShape and specify_shape in tensor

not tensor.basic
上级 91c7197c
...@@ -435,7 +435,7 @@ where, each of the optimization do the following things: ...@@ -435,7 +435,7 @@ where, each of the optimization do the following things:
acceptable_ops = (theano.tensor.basic.Dot, acceptable_ops = (theano.tensor.basic.Dot,
theano.tensor.basic.Reshape, theano.tensor.basic.Reshape,
theano.tensor.basic.Shape, theano.tensor.basic.Shape,
theano.tensor.basic.SpecifyShape, theano.tensor.SpecifyShape,
theano.tensor.basic.MaxAndArgmax, theano.tensor.basic.MaxAndArgmax,
theano.tensor.Subtensor, theano.tensor.Subtensor,
theano.tensor.IncSubtensor, theano.tensor.IncSubtensor,
......
...@@ -65,3 +65,6 @@ from theano.tensor.sort import sort, argsort ...@@ -65,3 +65,6 @@ from theano.tensor.sort import sort, argsort
from theano.tensor.extra_ops import (DiffOp, bincount, squeeze, from theano.tensor.extra_ops import (DiffOp, bincount, squeeze,
repeat, bartlett, fill_diagonal, fill_diagonal_offset, repeat, bartlett, fill_diagonal, fill_diagonal_offset,
cumsum, cumprod) cumsum, cumprod)
# SpecifyShape is defined in theano.compile, but should be available in tensor
from theano.compile import SpecifyShape, specify_shape
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论