NEML2 1.4.0
Loading...
Searching...
No Matches
WR2 Class Reference

Public Member Functions

WR2 __add__ (self, float arg0)
 
WR2 __add__ (self, Scalar arg0)
 
WR2 __add__ (self, WR2 arg0)
 
Scalar __call__ (self, int arg0, int arg1)
 
Scalar __call__ (self, int arg0, int arg1)
 
None __init__ (self)
 
None __init__ (self, torch.Tensor arg0, int arg1)
 
None __init__ (self, WR2 arg0)
 
None __init__ (self, torch.Tensor arg0)
 
None __init__ (self, R2 arg0)
 
WR2 __mul__ (self, float arg0)
 
WR2 __mul__ (self, Scalar arg0)
 
WR2 __neg__ (self)
 
WR2 __pow__ (self, float arg0)
 
WR2 __pow__ (self, Scalar arg0)
 
WR2 __radd__ (self, float arg0)
 
str __repr__ (self)
 
WR2 __rmul__ (self, float arg0)
 
Tensor __rpow__ (self, float arg0)
 
WR2 __rsub__ (self, float arg0)
 
WR2 __rtruediv__ (self, float arg0)
 
str __str__ (self)
 
WR2 __sub__ (self, float arg0)
 
WR2 __sub__ (self, Scalar arg0)
 
WR2 __sub__ (self, WR2 arg0)
 
WR2 __truediv__ (self, float arg0)
 
WR2 __truediv__ (self, Scalar arg0)
 
WR2 __truediv__ (self, WR2 arg0)
 
bool batched (self)
 
WR2 clone (self)
 
torch.Tensor copy_ (self, torch.Tensor arg0, bool arg1)
 
WR2 cross (self, Vec arg0)
 
WR2 cross (self, Rot arg0)
 
WR2 cross (self, WR2 arg0)
 
bool defined (self)
 
WR2 detach (self)
 
torch.Tensor detach_ (self)
 
R2 dexp (self)
 
int dim (self)
 
Scalar dot (self, Vec arg0)
 
Scalar dot (self, Rot arg0)
 
Scalar dot (self, WR2 arg0)
 
R2 drotate (self, Rot arg0)
 
Rot exp (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)
 
WR2 rotate (self, Rot arg0)
 
Tensor tensor (self)
 
WR2 to (self, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (self)
 
torch.Tensor torch (self)
 
torch.Tensor zero_ (self)
 
WR2BaseView base (self)
 
WR2BatchView 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

WR2 empty (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 empty (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 empty_like (WR2 arg0)
 
WR2 fill (float x, float y, float z, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 fill (Scalar x, Scalar y, Scalar z)
 
WR2 full (float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 full (tuple[int,...] batch_shape, float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 full_like (WR2 arg0, float arg1)
 
R2 identity_map (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 linspace (WR2 start, WR2 end, int nstep, int dim=0, int batch_dim=-1)
 
WR2 logspace (WR2 start, WR2 end, int nstep, int dim=0, int batch_dim=-1, float base=10.0)
 
WR2 ones (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 ones (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 ones_like (WR2 arg0)
 
WR2 zeros (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 zeros (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
WR2 zeros_like (WR2 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,
WR2 arg0 )

◆ __init__() [4/5]

None __init__ ( self,
torch.Tensor arg0 )

◆ __init__() [5/5]

None __init__ ( self,
R2 arg0 )

Member Function Documentation

◆ __add__() [1/3]

WR2 __add__ ( self,
float arg0 )

◆ __add__() [2/3]

WR2 __add__ ( self,
Scalar arg0 )

◆ __add__() [3/3]

WR2 __add__ ( self,
WR2 arg0 )

◆ __call__() [1/2]

Scalar __call__ ( self,
int arg0,
int arg1 )

◆ __call__() [2/2]

Scalar __call__ ( self,
int arg0,
int arg1 )

◆ __mul__() [1/2]

WR2 __mul__ ( self,
float arg0 )

◆ __mul__() [2/2]

WR2 __mul__ ( self,
Scalar arg0 )

◆ __neg__()

WR2 __neg__ ( self)

◆ __pow__() [1/2]

WR2 __pow__ ( self,
float arg0 )

◆ __pow__() [2/2]

WR2 __pow__ ( self,
Scalar arg0 )

◆ __radd__()

WR2 __radd__ ( self,
float arg0 )

◆ __repr__()

str __repr__ ( self)

◆ __rmul__()

WR2 __rmul__ ( self,
float arg0 )

◆ __rpow__()

Tensor __rpow__ ( self,
float arg0 )

◆ __rsub__()

WR2 __rsub__ ( self,
float arg0 )

◆ __rtruediv__()

WR2 __rtruediv__ ( self,
float arg0 )

◆ __str__()

str __str__ ( self)

◆ __sub__() [1/3]

WR2 __sub__ ( self,
float arg0 )

◆ __sub__() [2/3]

WR2 __sub__ ( self,
Scalar arg0 )

◆ __sub__() [3/3]

WR2 __sub__ ( self,
WR2 arg0 )

◆ __truediv__() [1/3]

WR2 __truediv__ ( self,
float arg0 )

◆ __truediv__() [2/3]

WR2 __truediv__ ( self,
Scalar arg0 )

◆ __truediv__() [3/3]

WR2 __truediv__ ( self,
WR2 arg0 )

◆ base()

WR2BaseView base ( self)

◆ batch()

WR2BatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

WR2 clone ( self)

◆ copy_()

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

◆ cross() [1/3]

WR2 cross ( self,
Rot arg0 )

◆ cross() [2/3]

WR2 cross ( self,
Vec arg0 )

◆ cross() [3/3]

WR2 cross ( self,
WR2 arg0 )

◆ defined()

bool defined ( self)

◆ detach()

WR2 detach ( self)

◆ detach_()

torch.Tensor detach_ ( self)

◆ device()

torch.device device ( self)

◆ dexp()

R2 dexp ( 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()

R2 drotate ( self,
Rot arg0 )

◆ dtype()

torch.dtype dtype ( self)

◆ empty() [1/2]

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

◆ empty() [2/2]

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

◆ empty_like()

WR2 empty_like ( WR2 arg0)
static

◆ exp()

Rot exp ( self)

◆ fill() [1/2]

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

◆ fill() [2/2]

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

◆ full() [1/2]

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

◆ full() [2/2]

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

◆ full_like()

WR2 full_like ( WR2 arg0,
float arg1 )
static

◆ grad()

torch.Tensor grad ( self)

◆ identity_map()

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

◆ linspace()

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

◆ logspace()

WR2 logspace ( WR2 start,
WR2 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]

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

◆ ones() [2/2]

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

◆ ones_like()

WR2 ones_like ( WR2 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()

WR2 rotate ( self,
Rot arg0 )

◆ shape()

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

◆ tensor()

Tensor tensor ( self)

◆ to()

WR2 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]

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

◆ zeros() [2/2]

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

◆ zeros_like()

WR2 zeros_like ( WR2 arg0)
static