Baremetal-NN
Baremetal-NN API documentation
|
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
out | the output tensor of shape (m, n) |
a | the input tensor of shape (m, k) |
b | the input tensor of shape (n, k) |