NEML2 2.0.0
All Classes Namespaces Functions Variables Pages
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)
 
WR2 __iadd__ (self, float arg0)
 
WR2 __imul__ (self, float arg0)
 
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 __isub__ (self, float arg0)
 
WR2 __itruediv__ (self, float 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)
 
torch.Tensor __truediv__ (self, WR2 arg0)
 
bool batched (self)
 
WR2 clone (self)
 
torch.Tensor copy_ (self, torch.Tensor arg0, bool arg1)
 
bool defined (self)
 
WR2 detach (self)
 
torch.Tensor detach_ (self)
 
R2 dexp (self)
 
int dim (self)
 
R2 drotate (self, Rot arg0)
 
Rot exp (self)
 
Number|torch.SymInt|torch.SymFloat|torch.SymBool item (self)
 
Scalar norm (self)
 
Scalar norm_sq (self)
 
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)
 
WR2 logspace (WR2 start, WR2 end, int nstep, int dim=0, 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 )

◆ __iadd__()

WR2 __iadd__ ( self,
float arg0 )

◆ __imul__()

WR2 __imul__ ( self,
float arg0 )

◆ __isub__()

WR2 __isub__ ( self,
float arg0 )

◆ __itruediv__()

WR2 __itruediv__ ( self,
float arg0 )

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

torch.Tensor __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 )

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

◆ 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

◆ item()

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

◆ linspace()

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

◆ logspace()

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

◆ 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