Algorithme en python sommant des matrices sous forme compacte

def addition_diagonale(A,B): #additionne des matrices entrees sous forme compactes
    Somme=[]

    if len(A) != len(B) and len(A[0]) != len(B[0]): # criteres sur les dimensions pour que l addition soit faisable. Si les critères ne sont pas respectes on renvoie une liste vide
        return []
   
    for j in range(len(A)): #A et B ont meme dimension
        Somme.append(A[j]+B[j])
    return Somme