X = np.random.normal(size=(10, 5, 3))
Y = np.random.normal(size=(10, 3, 5))
Z1 = matrix_multiply(X, Y)
Z2 = np.array([np.dot(X[k], Y[k]) for k in range(10)])
np.allclose(Z1, Z2)
найти точечный продукт подмассивов в numpy