提交 08fd8ae3 authored 作者: AndroidCloud's avatar AndroidCloud

adding updates

上级 8f3f254b
...@@ -29,7 +29,8 @@ from theano import compile, printing ...@@ -29,7 +29,8 @@ from theano import compile, printing
from theano.printing import pprint, min_informative_str from theano.printing import pprint, min_informative_str
# For history # For history
from theano.compile import Rebroadcast, Shape, shape from theano.compile import Rebroadcast, Shape, shape
from theano.scalar import Scalar from theano.scalar import int32
# We use these exceptions as well. # We use these exceptions as well.
import theano.scalar.sharedvar import theano.scalar.sharedvar
...@@ -4719,7 +4720,7 @@ class Reshape(Op): ...@@ -4719,7 +4720,7 @@ class Reshape(Op):
check_input = False check_input = False
__props__ = ("ndim",) __props__ = ("ndim",)
params_type = ParamsType(ndim=Scalar('int32')) params_type = ParamsType(ndim=int32)
# name does not participate because it doesn't affect computations # name does not participate because it doesn't affect computations
def __init__(self, ndim, name=None): def __init__(self, ndim, name=None):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论