NEML2 2.1.0
Loading...
Searching...
No Matches
StereographicProjection Class Reference

Detailed Description

Stereographic projection

Public Member Functions

 __init__ (self)
 __call__ (self, v)
 inverse (self, v)

Public Attributes

float limit = 1.0

Constructor & Destructor Documentation

◆ __init__()

__init__ ( self)

Member Function Documentation

◆ __call__()

__call__ ( self,
v )
Project from 3d points on sphere to 2d

Args:
    v (torch.tensor): tensor of shape (..., 3)

◆ inverse()

inverse ( self,
v )
Inverse of the stereographic projection

Args:
    v (torch.tensor): tensor of shape (..., 2)

Member Data Documentation

◆ limit

float limit = 1.0