Get size of numpy matrix