Skip to main content

complex_mul

burn_mamba::mamba3::step_constant

Function complex_mul 

Source 
fn complex_mul(
    ar: Tensor<3>,
    ai: Tensor<3>,
    br: Tensor<3>,
    bi: Tensor<3>,
) -> (Tensor<3>, Tensor<3>)
Expand description

Component-wise complex product (ar + i·ai)(br + i·bi).