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