diff --git a/src/numerical/linear_algebra.py b/src/numerical/linear_algebra.py index 0c56ca3..d224731 100644 --- a/src/numerical/linear_algebra.py +++ b/src/numerical/linear_algebra.py @@ -11,8 +11,7 @@ def numpy_matmul(A: np.ndarray, B: np.ndarray) -> np.ndarray: result = np.zeros((rows_A, cols_B)) for i in range(rows_A): for j in range(cols_B): - for k in range(cols_A): - result[i, j] += A[i, k] * B[k, j] + result[i, j] = np.dot(A[i, :], B[:, j]) return result