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

◆ NN_mm_t()

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

Performs a matrix multiplication with transposed B.

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

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)