torch.bitwise_right_shift — PyTorch 2.7 documentation (original) (raw)

torch.bitwise_right_shift(input, other, *, out=None) → Tensor¶

Computes the right arithmetic shift of input by other bits. The input tensor must be of integral type. This operator supportsbroadcasting to a common shape andtype promotion. In any case, if the value of the right operand is negative or is greater or equal to the number of bits in the promoted left operand, the behavior is undefined.

The operation applied is:

outi=inputi>>otheri\text{out}_i = \text{input}_i >> \text{other}_i

Parameters

• input (Tensor or Scalar) – the first input tensor
• other (Tensor or Scalar) – the second input tensor

Keyword Arguments

out (Tensor, optional) – the output tensor.

Example:

torch.bitwise_right_shift(torch.tensor([-2, -7, 31], dtype=torch.int8), torch.tensor([1, 0, 3], dtype=torch.int8)) tensor([-1, -7, 3], dtype=torch.int8)