未验证 提交 81c89dd6 编写于 作者: J JYChen 提交者: GitHub

fix right value is 0d and index is List/Tensor (#53225)

上级 21508090
......@@ -185,7 +185,8 @@ class SliceInfo:
for i in range(len(gather_tensor_shape)):
if not (
value_dims_bd[i] == gather_tensor_shape[i]
len(value_dims_bd) == 0
or value_dims_bd[i] == gather_tensor_shape[i]
or value_dims_bd[i] == 1
):
raise ValueError(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册