diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py index f4c7a0a89ac..99f096d024d 100644 --- a/torch/_torch_docs.py +++ b/torch/_torch_docs.py @@ -1377,8 +1377,9 @@ add_docstr(torch.bitwise_left_shift, bitwise_left_shift(input, other, *, out=None) -> Tensor Computes the left arithmetic shift of :attr:`input` by :attr:`other` bits. -The result will have the same dtype as :attr:`input`. The input tensor must -be of integral types. +The input tensor must be of integral type. This operator supports +:ref:`broadcasting to a common shape ` and +:ref:`type promotion `. The operation applied is: @@ -1403,8 +1404,9 @@ add_docstr(torch.bitwise_right_shift, bitwise_right_shift(input, other, *, out=None) -> Tensor Computes the right arithmetic shift of :attr:`input` by :attr:`other` bits. -The result will have the same dtype as :attr:`input`. The input tensor must -be of integral types. +The input tensor must be of integral type. This operator supports +:ref:`broadcasting to a common shape ` and +:ref:`type promotion `. The operation applied is: