Element-wise maximum of two arrays, like base::pmax().
Arguments
- lhs, rhs
(
arrayish)
Two inputs of the same data type and shape. Can be any data type. R values take the other operand's data type when it is in their data type category, and their default data type when neither operand has one. An R value outside the other operand's category is an error, as are two R values of different storage types.
Value
(arrayish)
Has the inputs' shape and data type.
StableHLO
Lowers to hlo_maximum(), specified under
maximum.
Examples
# two R values: both take an R double's default data type
prim_pmax(1, 5)
#> AnvlArray
#> 5
#> [ CPUf32{} ]
# the R value is built at the array's data type instead
prim_pmax(1, nv_scalar(5, "f64"))
#> AnvlArray
#> 5
#> [ CPUf64{} ]