Skip to main content

elementwise_binary

Function elementwise_binary 

Source
pub fn elementwise_binary<R: Runtime>(
    a: &RudaTensor<R>,
    modes_a: &[Mode],
    b: &RudaTensor<R>,
    modes_b: &[Mode],
    output_modes: &[Mode],
    combine: BinaryOp,
) -> Result<RudaTensor<R>>
Expand description

Named-axis broadcasting followed by a binary operation.