ClikaRT::ops::mv

function

mv()

Tensor mv(
    Tensor a,
    Tensor vec,
    OptionalTensor bias = {},
    std::optional<Activation> activation = std::nullopt
)

Matrix-vector multiply, with an optional fused bias and activation.

a [M, K] x vec [K] -> [M].

Parameters

• a: the matrix [M, K].
• vec: the vector [K].
• bias: optional additive bias, broadcastable to [M].
• activation: optional fused epilogue; absent = none.

Returns: [M].

Throws

• ClikaRT::Error: (INVALID_ARGUMENT) when a is not rank-2, vec not rank-1, or K disagrees.

Declared in ClikaRT/compute/ops.h, line 2794