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