Primitive Arithmetic Shift Right

Element-wise arithmetic right bit shift.

Usage

nvl_shift_right_arithmetic(lhs, rhs)

Arguments

lhs, rhs

(tensorish)
Tensorish values of data type boolean, integer, or unsigned integer. Must have the same shape.

Value

tensorish
Has the same shape and data type as the inputs. It is ambiguous if both inputs are ambiguous.

Implemented Rules

• stablehlo

• backward

StableHLO

Lowers to stablehlo::hlo_shift_right_arithmetic().

See also

nv_shift_right_arithmetic()

Examples

jit_eval({
  x <- nv_tensor(c(8L, -16L, 32L))
  y <- nv_tensor(c(1L, 2L, 3L))
  nvl_shift_right_arithmetic(x, y)
})
#> AnvilTensor
#>   4
#>  -4
#>   4
#> [ CPUi32{3} ]