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

◆ NN_matmul_t()

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

Matrix multiplication with transposed B

C = A @ B.T

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