make equation behind torch.isclose element-wise

vladimirrotariu · vladimirrotariu · commit 075b2320ecc2 · 2024-10-21T15:19:19.000+02:00
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -5329,7 +5329,7 @@ def merge_dicts(*dicts):
 Closeness is defined as:
 
 .. math::
-    \lvert \text{input} - \text{other} \rvert \leq \texttt{atol} + \texttt{rtol} \times \lvert \text{other} \rvert
+    \lvert \text{input}_i - \text{other}_i \rvert \leq \texttt{atol} + \texttt{rtol} \times \lvert \text{other}_i \rvert \quad \forall i
 """
     + r"""
 
