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

◆ NN_matmul()

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

Matrix multiplication

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)