Skip to main content

int_binary_op

Function int_binary_op 

Source
pub fn int_binary_op<Op>(lhs: HostTensor, rhs: HostTensor, op: Op) -> HostTensor
where Op: Fn(i64, i64) -> i64 + Copy,
Expand description

Apply a binary operation element-wise to two integer tensors.

Supports all integer dtypes: I64, I32, I16, I8, U64, U32, U16, U8.