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

◆ NN_mm()

void NN_mm ( Tensor out,
const Tensor a,
const Tensor b 
)

Performs a matrix multiplication.

If input is a (m×k) tensor, mat2 is a (k×n) tensor, out will be a (m×n) tensor.

C = A @ B

Parameters
outthe output tensor of shape (m, n)
athe input tensor of shape (m, k)
bthe input tensor of shape (k, n)