Function gemm_ctx

Source
pub fn gemm_ctx<F: BLAS3<C> + Zero, C: BLASContext<Device = D>, D: Device>(
    ctx: C,
    alpha: F,
    a: MatrixView<'_, F, D>,
    b: MatrixView<'_, F, D>,
    beta: F,
    c: MatrixViewMut<'_, F, D>,
)
Expand description

Performs the basic matmul operation.

C = alpha * AB + beta * C.

ยงPanics

If the shapes of the matricies do not match the following pattern:

  • A = (M, K)
  • B = (K, N)
  • C = (M, N)