提交 9b9b76d7 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

fix a NotScalarConstantError issue

上级 268cf9cd
......@@ -1615,7 +1615,7 @@ def local_gemm_to_ger(node):
yv = y.dimshuffle(1)
try:
bval = T.get_scalar_constant_value(b)
except TypeError:
except T.NotScalarConstantError:
# b isn't a constant, GEMM is doing useful pre-scaling
return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论