Rate this Page

torch.bitwise_right_shift#

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

Computes the right arithmetic shift ofinput byother 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 greateror 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 orScalar) – the first input tensor

  • other (Tensor orScalar) – 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)