Update tensor docs

soulitzer · soulitzer · commit c948f1d1cfdb · 2021-01-20T21:44:33.000-05:00
diff --git a/torch/_tensor_docs.py b/torch/_tensor_docs.py
@@ -1160,7 +1160,7 @@ def add_docstr_all(method, docstr):
 
 add_docstr_all('diff',
                r"""
-diff(n=1, dim=-1) -> Tensor
+diff(n=1, dim=-1, prepend=None, append=None) -> Tensor
 
 See :func:`torch.diff`
 """)
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -2639,7 +2639,7 @@ def merge_dicts(*dicts):
 """.format(**common_args))
 
 add_docstr(torch.diff, r"""
-diff(input, *, n=1, dim=-1) -> Tensor
+diff(input, n=1, dim=-1, prepend=None, append=None) -> Tensor
 
 Computes the n-th forward difference along the given dimension.
 
