提交 475fbeb5 authored 作者: Sigurd Spieckermann's avatar Sigurd Spieckermann

fixed case where shape cannot be inferred

上级 489b04e4
...@@ -1992,7 +1992,7 @@ def local_useless_subtensor(node): ...@@ -1992,7 +1992,7 @@ def local_useless_subtensor(node):
try: try:
length = get_scalar_constant_value(shape_of[node.inputs[0]][0]) length = get_scalar_constant_value(shape_of[node.inputs[0]][0])
except NotScalarConstantError: except NotScalarConstantError:
pass return False
# get index (which must be a vector by definition) # get index (which must be a vector by definition)
idx = node.inputs[1] idx = node.inputs[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论