Calcule la suma de todos los elementos de la matriz de 10 x 10 Con numpy

import numpy as np
def sum_matrix(matrix):
  return np.sum(matrix, axis=0)

matrix = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(sum_matrix(matrix))

A:

You can use np.einsum to do this. It is a very powerful tool for doing matrix