Add docs for tensor.index_put_ (#6563)
diff --git a/docs/source/tensors.rst b/docs/source/tensors.rst
index 7fbf970..a1f8e10 100644
--- a/docs/source/tensors.rst
+++ b/docs/source/tensors.rst
@@ -187,6 +187,7 @@
.. automethod:: index_add_
.. automethod:: index_copy_
.. automethod:: index_fill_
+ .. automethod:: index_put_
.. automethod:: index_select
.. automethod:: int
.. automethod:: inverse
diff --git a/torch/_tensor_docs.py b/torch/_tensor_docs.py
index 349e874..950c826 100644
--- a/torch/_tensor_docs.py
+++ b/torch/_tensor_docs.py
@@ -749,6 +749,20 @@
""")
+add_docstr_all('index_put_',
+ r"""
+index_put_(indices, value) -> Tensor
+
+Puts values from the tensor :attr:`value` into the tensor :attr:`self` using
+the indices specified in :attr:`indices` (which is a tuple of Tensors). The
+expression ``tensor.index_put_(indices, value)`` is equivalent to
+``tensor[indices] = value``. Returns :attr:`self`.
+
+Args:
+ indices (tuple of LongTensor): tensors used to index into `self`.
+ value (Tensor): tensor of same dtype as `self`.
+""")
+
add_docstr_all('index_select',
r"""
index_select(dim, index) -> Tensor
