diff --git a/einops/_backends.py b/einops/_backends.py index db43cb17..b18a60b0 100644 --- a/einops/_backends.py +++ b/einops/_backends.py @@ -623,7 +623,7 @@ def __init__(self): self.paddle = paddle def is_appropriate_type(self, tensor): - return isinstance(tensor, (self.paddle.Tensor, self.paddle.static.Variable)) + return self.paddle.is_tensor(tensor) def from_numpy(self, x): tensor = self.paddle.to_tensor(x)