Torch Unique Return Inverse Tangent