Implement something similar to this one : https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html on PMMatrix