Function tract_hir::internal::tract_ndarray::linalg::general_mat_vec_mul[][src]

pub fn general_mat_vec_mul<A, S1, S2, S3>(
    alpha: A,
    a: &ArrayBase<S1, Dim<[usize; 2]>>,
    x: &ArrayBase<S2, Dim<[usize; 1]>>,
    beta: A,
    y: &mut ArrayBase<S3, Dim<[usize; 1]>>
) where
    A: LinalgScalar,
    S2: Data<Elem = A>,
    S1: Data<Elem = A>,
    S3: DataMut<Elem = A>, 

General matrix-vector multiplication.

Compute y ← α A x + β y

where A is a M × N matrix and x is an N-element column vector and y an M-element column vector (one dimensional arrays).

Panics if array shapes are not compatible
Note: If enabled, uses blas gemv for elements of f32, f64 when memory layout allows.