From f57392b94f22f31c3489d4f99507b0226dbe0853 Mon Sep 17 00:00:00 2001 From: Guan-t7 Date: Wed, 8 Nov 2023 12:32:55 -0800 Subject: [PATCH] Fix __rsub__ in Tensor --- python/needle/autograd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/needle/autograd.py b/python/needle/autograd.py index 60ea47a..50ba4bc 100755 --- a/python/needle/autograd.py +++ b/python/needle/autograd.py @@ -360,8 +360,9 @@ def transpose(self, axes=None): __radd__ = __add__ __rmul__ = __mul__ - __rsub__ = __sub__ __rmatmul__ = __matmul__ + def __rsub__(self, other): + return other + (-self) def compute_gradient_of_variables(output_tensor, out_grad):