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
out
the output tensor of shape (m, n)
a
the input tensor of shape (m, k)
b
the input tensor of shape (n, k)
nn
functional
matmul.h
Generated by
1.9.5