Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches

◆ NN_mv()

void NN_mv ( Tensor out,
const Tensor a,
const Tensor v 
)

Performs a matrix multiplication.

If input is a (n, m) tensor, v is a (m, ) tensor, out will be a (n, ) tensor.

C = A @ B

Parameters
outthe output tensor of shape (n,)
athe input tensor of shape (n, m)
vthe input vector tensor of shape (m,)