pub fn matmul_2d(lhs: &Tensor, rhs: &Tensor) -> Result<Tensor, KernelError>
Deterministic rank-2 matrix multiplication: (m x k) * (k x n) -> (m x n).
(m x k) * (k x n) -> (m x n)