提交 bae722ec 编写于 作者: W wjj19950828

fixed import models bug

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