remove unnecessary torch.Tensor.mul override

samuela · Nov 7, 2023 · 97b9722 · 97b9722
1 parent 1446b83
commit 97b9722
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torch2jax/__init__.py b/torch2jax/__init__.py
@@ -170,8 +170,6 @@ def fn(*args, **kwargs):
 connect(torch.tanh, jnp.tanh)
 connect(torch.transpose, jnp.swapaxes)
 
-connect(torch.Tensor.mul, jnp.multiply)
-
 
 @implements(torch.cat)
 def cat(tensors, dim=0):