I plot a matrix ‘diff’ (quadratique sum of the input matrix diff_X and diff_Y) as a colored map using imshow and I would like to display arrows showing the corresponding angle arctan2(diff_Y/diff_X) for each coordinates. This is my program so far: