Skip to content

Commit 065c598

Browse files
gchanancolesbury
authored andcommitted
Fix docs: masked_fill_ takes a value, not a tensor. (pytorch#1663)
1 parent 45f665d commit 065c598

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/_tensor_docs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -827,7 +827,7 @@ def callable(a, b) -> number
827827
828828
Args:
829829
mask (ByteTensor): The binary mask
830-
value (Tensor): The value to fill
830+
value (float): The value to fill
831831
""")
832832

833833
add_docstr(torch._C.FloatTensorBase.masked_select,

0 commit comments

Comments
 (0)