Skip to main content

binary_op

tenflowers_core::ops::binary::implementation

Function binary_op 

Source 
pub fn binary_op<T, Op>(
    a: &Tensor<T>,
    b: &Tensor<T>,
    op: Op,
) -> Result<Tensor<T>>
where
    T: Clone + Default + Zero + Send + Sync + 'static + Pod + Zeroable,
    Op: BinaryOp<T>,
Expand description

Generic binary operation implementation with broadcasting