提交 bae722ec 编写于 作者: W wjj19950828

fixed import models bug

上级 b220a63e
......@@ -24,7 +24,7 @@ def is_condition_one(idx):
a[mask, :]
a[mask, ...]
"""
if not (isinstance(idx[0], paddle.Tensor) and \
if not (isinstance(idx[0], paddle.Tensor) and
idx[0].dtype == paddle_dtypes.t_bool):
return False
if len(idx) == 1:
......@@ -94,6 +94,7 @@ def __getitem__(self, idx):
else:
return out
VarBase = core.eager.Tensor
VarBase.__getitem__ = __getitem__
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册