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