yobx.helpers._einsum.blas_lapack#
- yobx.helpers._einsum.blas_lapack.gemm_dot(A: ndarray, B: ndarray, transA: bool = False, transB: bool = False) ndarray[source]#
Implements a dot product using numpy matmul.
- Parameters:
A – first matrix (2-D)
B – second matrix (2-D)
transA – transposes A before the multiplication
transB – transposes B before the multiplication
- Returns:
output matrix