Add instance methods tensor.isnan(), tensor.isinf(), tensor.isfinite()

ShawnZhong · ShawnZhong · commit 341fb19cf46c · 2020-05-08T09:55:38.000-05:00
Update _tensor_docs.py

Update tensors.rst

Update _tensor_docs.py
diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
@@ -1586,7 +1586,7 @@
 
 - func: isnan(Tensor self) -> Tensor
   use_c10_dispatcher: full
-  variants: function
+  variants: function, method
   device_guard: False
   supports_named_tensor: True
   dispatch:
@@ -6973,13 +6973,13 @@
 
 - func: isfinite(Tensor self) -> Tensor
   use_c10_dispatcher: full
-  variants: function
+  variants: function, method
   device_guard: False
   supports_named_tensor: True
 
 - func: isinf(Tensor self) -> Tensor
   use_c10_dispatcher: full
-  variants: function
+  variants: function, method
   device_guard: False
   supports_named_tensor: True
 
diff --git a/docs/source/tensors.rst b/docs/source/tensors.rst
@@ -313,6 +313,9 @@ view of a storage and defines numeric operations on it.
    .. automethod:: inverse
    .. automethod:: irfft
    .. automethod:: isclose
+   .. automethod:: isfinite
+   .. automethod:: isinf
+   .. automethod:: isnan
    .. automethod:: is_contiguous
    .. automethod:: is_complex
    .. automethod:: is_floating_point
diff --git a/torch/_tensor_docs.py b/torch/_tensor_docs.py
@@ -1541,6 +1541,27 @@ def add_docstr_all(method, docstr):
 See :func:`torch.inverse`
 """)
 
+add_docstr_all('isnan',
+               r"""
+isnan() -> Tensor
+
+See :func:`torch.isnan`
+""")
+
+add_docstr_all('isinf',
+               r"""
+isinf() -> Tensor
+
+See :func:`torch.isinf`
+""")
+
+add_docstr_all('isfinite',
+               r"""
+isfinite() -> Tensor
+
+See :func:`torch.isfinite`
+""")
+
 add_docstr_all('isclose',
                r"""
 isclose(other, rtol=1e-05, atol=1e-08, equal_nan=False) -> Tensor
