Esta función recibe tres array de Numpy de 2 dimensiones cada uno, y devuelve el valor booleano True si es posible realizar una multiplicación entre las tres matrices (n1 x n2 x n3), y el valor booleano False si no lo es.

import numpy as np

def mult(a, b, c):
    return a.shape[1] == b.shape[0] and b.shape[1] == c.shape[0]