Function addmm_device

pub fn addmm_device(
    c: impl AsRef<Array>,
    a: impl AsRef<Array>,
    b: impl AsRef<Array>,
    alpha: impl Into<Option<f32>>,
    beta: impl Into<Option<f32>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>

Matrix multiplication with addition and optional scaling.

Perform the (possibly batched) matrix multiplication of two arrays and add to the result with optional scaling factors.

Params

• c: input array,
• a: input array,
• b: input array,
• alpha: Scaling factor for the matrix product of a and b (default: 1)
• beta: Scaling factor for c (default: 1)