Código Python "simplificado" y Cython de la multiplicación de dos matrices

[delapuente]

En el tema de "Python de alto rendimiento", el último ejercicio pide que quitéis los generadores, prefiriendo el acceso directo a las matrices. Esta es una simplificación que cumple estas preferencias:

def mul(a, b):
    size = len(a)
    indices = [i for i in range(size)]
    result = [[0 for _ in indices] for _ in indices]
    for a_row_index in indices:
        for b_col_index in indices:
            item = 0
            for index in indices:
                item += a[a_row_index][index] * b[index][b_col_index]            
            
            result[a_row_index][b_col_index] = item

    return result

[delapuente]

Aquí el código de Cython para el que hemos encontrado mayor incremento de la velocidad:

def mul(long[:, :] a, long[:, :] b):
    cdef long i, _, a_row_index, b_col_index, index
    cdef long size, item
    
    size = len(a)
    indices = [i for i in range(size)]
    result = [[0 for _ in indices] for _ in indices]
    for a_row_index in indices:
        for b_col_index in indices:
            item = 0
            for index in indices:
                item += a[a_row_index][index] * b[index][b_col_index]            
            
            result[a_row_index][b_col_index] = item

    return result