NEML2 1.4.0
Loading...
Searching...
No Matches
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)
 
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 __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)
 
Rot __truediv__ (self, Rot arg0)
 
bool batched (self)
 
Rot clone (self)
 
torch.Tensor copy_ (self, torch.Tensor arg0, bool arg1)
 
Rot cross (self, Vec arg0)
 
Rot cross (self, Rot arg0)
 
Rot cross (self, WR2 arg0)
 
bool defined (self)
 
Rot detach (self)
 
torch.Tensor detach_ (self)
 
R3 deuler_rodrigues (self)
 
int dim (self)
 
Scalar dot (self, Vec arg0)
 
Scalar dot (self, Rot arg0)
 
Scalar dot (self, WR2 arg0)
 
R2 drotate (self, Rot arg0)
 
R2 drotate (self, Rot arg0)
 
R2 euler_rodrigues (self)
 
Rot inverse (self)
 
Scalar norm (self)
 
Scalar norm_sq (self)
 
R2 outer (self, Vec arg0)
 
R2 outer (self, Rot arg0)
 
R2 outer (self, WR2 arg0)
 
torch.Tensor requires_grad_ (self, bool arg0)
 
Rot rotate (self, Rot arg0)
 
Rot rotate (self, Rot arg0)
 
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 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, int batch_dim=-1)
 
Rot logspace (Rot start, Rot end, int nstep, int dim=0, int batch_dim=-1, 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 )

◆ __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]

Rot __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 )

◆ cross() [1/3]

Rot cross ( self,
Rot arg0 )

◆ cross() [2/3]

Rot cross ( self,
Vec arg0 )

◆ cross() [3/3]

Rot cross ( self,
WR2 arg0 )

◆ 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)

◆ dot() [1/3]

Scalar dot ( self,
Rot arg0 )

◆ dot() [2/3]

Scalar dot ( self,
Vec arg0 )

◆ dot() [3/3]

Scalar dot ( self,
WR2 arg0 )

◆ drotate() [1/2]

R2 drotate ( self,
Rot arg0 )

◆ drotate() [2/2]

R2 drotate ( self,
Rot arg0 )

◆ dtype()

torch.dtype dtype ( 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

◆ 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)

◆ linspace()

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

◆ logspace()

Rot logspace ( Rot start,
Rot end,
int nstep,
int dim = 0,
int batch_dim = -1,
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

◆ outer() [1/3]

R2 outer ( self,
Rot arg0 )

◆ outer() [2/3]

R2 outer ( self,
Vec arg0 )

◆ outer() [3/3]

R2 outer ( self,
WR2 arg0 )

◆ 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 )

◆ 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