提交 2d52b5f1 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

changed another except TypeError

上级 e10365de
...@@ -2686,7 +2686,7 @@ class GpuAlloc(GpuOp): ...@@ -2686,7 +2686,7 @@ class GpuAlloc(GpuOp):
# if s is constant 1, then we're broadcastable in that dim # if s is constant 1, then we're broadcastable in that dim
try: try:
const_shp = tensor.get_scalar_constant_value(s) const_shp = tensor.get_scalar_constant_value(s)
except TypeError: except tensor.NotScalarConstantError:
const_shp = None const_shp = None
bcast.append(numpy.all(1 == const_shp)) bcast.append(numpy.all(1 == const_shp))
otype = CudaNdarrayType(dtype='float32', broadcastable=bcast) otype = CudaNdarrayType(dtype='float32', broadcastable=bcast)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论