Add doc for Tensor.digamma_? (#11008)

ssnl · facebook-github-bot · commit e9eed8edb438 · 2018-08-29T14:11:16.000-07:00
Summary: follow up for #10967 zou3519 vishwakftw Pull Request resolved: #11008 Differential Revision: D9559889 Pulled By: SsnL fbshipit-source-id: a05d8fbad92a54bcdb93de6e62a7f94180da1d99
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -179,8 +179,6 @@ def test_namespace(ns, *skips):
                        'as_strided_',
                        re.compile('^clamp_(min|max)_?$'),
                        'coalesce',
-                       'digamma',
-                       'digamma_',
                        'index_put',
                        'is_coalesced',
                        'is_distributed',
diff --git a/torch/_tensor_docs.py b/torch/_tensor_docs.py
@@ -650,6 +650,20 @@ def add_docstr_all(method, docstr):
 See :func:`torch.diagonal`
 """)
 
+add_docstr_all('digamma',
+               r"""
+digamma() -> Tensor
+
+See :func:`torch.digamma`
+""")
+
+add_docstr_all('digamma_',
+               r"""
+digamma_() -> Tensor
+
+In-place version of :meth:`~Tensor.digamma`
+""")
+
 add_docstr_all('dim',
                r"""
 dim() -> int
