Baremetal-NN
Baremetal-NN API documentation
|
Performs a matrix multiplication.
If input is a (n, m) tensor, v is a (m, ) tensor, out will be a (n, ) tensor.
C = A @ B
out | the output tensor of shape (n,) |
a | the input tensor of shape (n, m) |
v | the input vector tensor of shape (m,) |