Primitive Arithmetic Shift Right

Element-wise arithmetic right bit shift.

Usage

prim_shift_right_arithmetic(x, shift)

Arguments

x

(arrayish)
The array whose bits are shifted. Can be any integer data type.

shift

(arrayish)
By how many bits to shift each element of x. Has the same data type and shape as x. An R value takes the other operand's data type, and its default data type when neither operand has one.

Value

(arrayish)
Has the inputs' shape and data type.

Implemented Rules

• stablehlo

• reverse

StableHLO

Lowers to hlo_shift_right_arithmetic(), specified under shift_right_arithmetic.

See also

nv_shift_right_arithmetic()

Examples

# two R values: both take an R integer's default data type
prim_shift_right_arithmetic(-32L, 2L)
#> AnvlArray
#>  -8
#> [ CPUi32{} ] 

# the R value is built at the array's data type instead
prim_shift_right_arithmetic(-32L, nv_scalar(2L, "i64"))
#> AnvlArray
#>  -8
#> [ CPUi64{} ]