Gradient magnitude for gradient jacobian tensor

Hi ,
i have jacobian gradient matrix for output image w.r.t input image and i try to calculate the greatest gradient magnitude (horizontal and vertical ) for this jacobian gradient matrix (tensor).
Is there any function in Pytorch calulate gradient magnitude for a tensor?

Thank you