NEML2 2.0.0
All Classes Namespaces Functions Variables Pages
Rot Class Reference

Public Member Functions

Rot __add__ (self, float arg0)
 
Rot __add__ (self, Scalar arg0)
 
Rot __add__ (self, Rot arg0)
 
Scalar __call__ (self, int arg0)
 
Rot __iadd__ (self, float arg0)
 
Rot __imul__ (self, float arg0)
 
None __init__ (self)
 
None __init__ (self, torch.Tensor arg0, int arg1)
 
None __init__ (self, Rot arg0)
 
None __init__ (self, torch.Tensor arg0)
 
None __init__ (self, Vec arg0)
 
Rot __isub__ (self, float arg0)
 
Rot __itruediv__ (self, float arg0)
 
Rot __mul__ (self, float arg0)
 
Rot __mul__ (self, Scalar arg0)
 
Rot __mul__ (self, Rot arg0)
 
Rot __neg__ (self)
 
Rot __pow__ (self, float arg0)
 
Rot __pow__ (self, Scalar arg0)
 
Rot __radd__ (self, float arg0)
 
str __repr__ (self)
 
Rot __rmul__ (self, float arg0)
 
Tensor __rpow__ (self, float arg0)
 
Rot __rsub__ (self, float arg0)
 
Rot __rtruediv__ (self, float arg0)
 
str __str__ (self)
 
Rot __sub__ (self, float arg0)
 
Rot __sub__ (self, Scalar arg0)
 
Rot __sub__ (self, Rot arg0)
 
Rot __truediv__ (self, float arg0)
 
Rot __truediv__ (self, Scalar arg0)
 
torch.Tensor __truediv__ (self, Rot arg0)
 
bool batched (self)
 
Rot clone (self)
 
torch.Tensor copy_ (self, torch.Tensor arg0, bool arg1)
 
Scalar dV (self)
 
bool defined (self)
 
Rot detach (self)
 
torch.Tensor detach_ (self)
 
R3 deuler_rodrigues (self)
 
int dim (self)
 
Scalar dist (self, Rot arg0)
 
R2 drotate (self, Rot arg0)
 
R2 drotate (self, Rot arg0)
 
R2 euler_rodrigues (self)
 
Rot inverse (self)
 
Number|torch.SymInt|torch.SymFloat|torch.SymBool item (self)
 
Scalar norm (self)
 
Scalar norm_sq (self)
 
torch.Tensor requires_grad_ (self, bool arg0)
 
Rot rotate (self, Rot arg0)
 
Rot rotate (self, Rot arg0)
 
Rot shadow (self)
 
Tensor tensor (self)
 
Rot to (self, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (self)
 
torch.Tensor torch (self)
 
torch.Tensor zero_ (self)
 
RotBaseView base (self)
 
RotBatchView batch (self)
 
torch.device device (self)
 
torch.dtype dtype (self)
 
torch.Tensor grad (self)
 
bool requires_grad (self)
 
tuple[int,...] shape (self)
 

Static Public Member Functions

Rot empty (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot empty (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot empty_like (Rot arg0)
 
Rot fill (float x, float y, float z, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot fill (Scalar x, Scalar y, Scalar z)
 
Rot fill_euler_angles (Vec arg0, str arg1, str arg2)
 
Rot fill_matrix (R2 arg0)
 
Rot fill_random (int arg0)
 
Rot fill_rodrigues (Scalar arg0, Scalar arg1, Scalar arg2)
 
Rot full (float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot full (tuple[int,...] batch_shape, float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot full_like (Rot arg0, float arg1)
 
Rot identity (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 identity_map (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot linspace (Rot start, Rot end, int nstep, int dim=0)
 
Rot logspace (Rot start, Rot end, int nstep, int dim=0, float base=10.0)
 
Rot ones (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot ones (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot ones_like (Rot arg0)
 
Rot zeros (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot zeros (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Rot zeros_like (Rot arg0)
 

Constructor & Destructor Documentation

◆ __init__() [1/5]

None __init__ ( self)

◆ __init__() [2/5]

None __init__ ( self,
torch.Tensor arg0,
int arg1 )

◆ __init__() [3/5]

None __init__ ( self,
Rot arg0 )

◆ __init__() [4/5]

None __init__ ( self,
torch.Tensor arg0 )

◆ __init__() [5/5]

None __init__ ( self,
Vec arg0 )

Member Function Documentation

◆ __add__() [1/3]

Rot __add__ ( self,
float arg0 )

◆ __add__() [2/3]

Rot __add__ ( self,
Rot arg0 )

◆ __add__() [3/3]

Rot __add__ ( self,
Scalar arg0 )

◆ __call__()

Scalar __call__ ( self,
int arg0 )

◆ __iadd__()

Rot __iadd__ ( self,
float arg0 )

◆ __imul__()

Rot __imul__ ( self,
float arg0 )

◆ __isub__()

Rot __isub__ ( self,
float arg0 )

◆ __itruediv__()

Rot __itruediv__ ( self,
float arg0 )

◆ __mul__() [1/3]

Rot __mul__ ( self,
float arg0 )

◆ __mul__() [2/3]

Rot __mul__ ( self,
Rot arg0 )

◆ __mul__() [3/3]

Rot __mul__ ( self,
Scalar arg0 )

◆ __neg__()

Rot __neg__ ( self)

◆ __pow__() [1/2]

Rot __pow__ ( self,
float arg0 )

◆ __pow__() [2/2]

Rot __pow__ ( self,
Scalar arg0 )

◆ __radd__()

Rot __radd__ ( self,
float arg0 )

◆ __repr__()

str __repr__ ( self)

◆ __rmul__()

Rot __rmul__ ( self,
float arg0 )

◆ __rpow__()

Tensor __rpow__ ( self,
float arg0 )

◆ __rsub__()

Rot __rsub__ ( self,
float arg0 )

◆ __rtruediv__()

Rot __rtruediv__ ( self,
float arg0 )

◆ __str__()

str __str__ ( self)

◆ __sub__() [1/3]

Rot __sub__ ( self,
float arg0 )

◆ __sub__() [2/3]

Rot __sub__ ( self,
Rot arg0 )

◆ __sub__() [3/3]

Rot __sub__ ( self,
Scalar arg0 )

◆ __truediv__() [1/3]

Rot __truediv__ ( self,
float arg0 )

◆ __truediv__() [2/3]

torch.Tensor __truediv__ ( self,
Rot arg0 )

◆ __truediv__() [3/3]

Rot __truediv__ ( self,
Scalar arg0 )

◆ base()

RotBaseView base ( self)

◆ batch()

RotBatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

Rot clone ( self)

◆ copy_()

torch.Tensor copy_ ( self,
torch.Tensor arg0,
bool arg1 )

◆ defined()

bool defined ( self)

◆ detach()

Rot detach ( self)

◆ detach_()

torch.Tensor detach_ ( self)

◆ deuler_rodrigues()

R3 deuler_rodrigues ( self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( self)

◆ dist()

Scalar dist ( self,
Rot arg0 )

◆ drotate() [1/2]

R2 drotate ( self,
Rot arg0 )

◆ drotate() [2/2]

R2 drotate ( self,
Rot arg0 )

◆ dtype()

torch.dtype dtype ( self)

◆ dV()

Scalar dV ( self)

◆ empty() [1/2]

Rot empty ( * ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ empty() [2/2]

Rot empty ( tuple[int, ...] batch_shape,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ empty_like()

Rot empty_like ( Rot arg0)
static

◆ euler_rodrigues()

R2 euler_rodrigues ( self)

◆ fill() [1/2]

Rot fill ( float x,
float y,
float z,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ fill() [2/2]

Rot fill ( Scalar x,
Scalar y,
Scalar z )
static

◆ fill_euler_angles()

Rot fill_euler_angles ( Vec arg0,
str arg1,
str arg2 )
static

◆ fill_matrix()

Rot fill_matrix ( R2 arg0)
static

◆ fill_random()

Rot fill_random ( int arg0)
static

◆ fill_rodrigues()

Rot fill_rodrigues ( Scalar arg0,
Scalar arg1,
Scalar arg2 )
static

◆ full() [1/2]

Rot full ( float fill_value,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ full() [2/2]

Rot full ( tuple[int, ...] batch_shape,
float fill_value,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ full_like()

Rot full_like ( Rot arg0,
float arg1 )
static

◆ grad()

torch.Tensor grad ( self)

◆ identity()

Rot identity ( * ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ identity_map()

R2 identity_map ( * ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ inverse()

Rot inverse ( self)

◆ item()

Number | torch.SymInt | torch.SymFloat | torch.SymBool item ( self)

◆ linspace()

Rot linspace ( Rot start,
Rot end,
int nstep,
int dim = 0 )
static

◆ logspace()

Rot logspace ( Rot start,
Rot end,
int nstep,
int dim = 0,
float base = 10.0 )
static

◆ norm()

Scalar norm ( self)

◆ norm_sq()

Scalar norm_sq ( self)

◆ ones() [1/2]

Rot ones ( * ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ ones() [2/2]

Rot ones ( tuple[int, ...] batch_shape,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ ones_like()

Rot ones_like ( Rot arg0)
static

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

torch.Tensor requires_grad_ ( self,
bool arg0 )

◆ rotate() [1/2]

Rot rotate ( self,
Rot arg0 )

◆ rotate() [2/2]

Rot rotate ( self,
Rot arg0 )

◆ shadow()

Rot shadow ( self)

◆ shape()

tuple[int, ...] shape ( self)

◆ tensor()

Tensor tensor ( self)

◆ to()

Rot to ( self,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )

◆ torch() [1/2]

torch.Tensor torch ( self)

◆ torch() [2/2]

torch.Tensor torch ( self)

◆ zero_()

torch.Tensor zero_ ( self)

◆ zeros() [1/2]

Rot zeros ( * ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ zeros() [2/2]

Rot zeros ( tuple[int, ...] batch_shape,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ zeros_like()

Rot zeros_like ( Rot arg0)
static