NEML2 2.0.0
Loading...
Searching...
No Matches
R2 Class Reference

Public Member Functions

R2 __add__ (R2 self, R2 arg0)
 
R2 __add__ (R2 self, Scalar arg0)
 
R2 __add__ (R2 self, float arg0)
 
R2 __iadd__ (R2 self, float arg0)
 
R2 __imul__ (R2 self, float arg0)
 
None __init__ (self)
 
None __init__ (self, torch.Tensor tensor, int dynamic_dim, int intmd_dim)
 
None __init__ (self, Vec other)
 
None __init__ (self, Rot other)
 
None __init__ (self, WR2 other)
 
None __init__ (self, R2 other)
 
None __init__ (self, SR2 other)
 
None __init__ (self, R3 other)
 
None __init__ (self, SFR3 other)
 
None __init__ (self, R4 other)
 
None __init__ (self, SFFR4 other)
 
None __init__ (self, WFFR4 other)
 
None __init__ (self, SSR4 other)
 
None __init__ (self, SWR4 other)
 
None __init__ (self, WSR4 other)
 
None __init__ (self, WWR4 other)
 
None __init__ (self, Quaternion other)
 
None __init__ (self, MillerIndex other)
 
None __init__ (self, Tensor other)
 
None __init__ (self, Scalar other)
 
None __init__ (self, torch.Tensor tensor, int intmd_dim=0)
 
R2 __isub__ (R2 self, float arg0)
 
R2 __itruediv__ (R2 self, float arg0)
 
R2 __mul__ (R2 self, R2 arg0)
 
Vec __mul__ (R2 self, Vec arg0)
 
R2 __mul__ (R2 self, Scalar arg0)
 
R2 __mul__ (R2 self, float arg0)
 
R2 __neg__ (R2 self)
 
R2 __pow__ (R2 self, float arg0)
 
R2 __radd__ (R2 self, Scalar arg0)
 
R2 __radd__ (R2 self, float arg0)
 
str __repr__ (R2 self)
 
R2 __rmul__ (R2 self, Scalar arg0)
 
R2 __rmul__ (R2 self, float arg0)
 
R2 __rsub__ (R2 self, Scalar arg0)
 
R2 __rsub__ (R2 self, float arg0)
 
R2 __rtruediv__ (R2 self, Scalar arg0)
 
R2 __rtruediv__ (R2 self, float arg0)
 
str __str__ (R2 self)
 
R2 __sub__ (R2 self, R2 arg0)
 
R2 __sub__ (R2 self, Scalar arg0)
 
R2 __sub__ (R2 self, float arg0)
 
R2 __truediv__ (R2 self, Scalar arg0)
 
R2 __truediv__ (R2 self, float arg0)
 
R2 clone (R2 self)
 
R2 contiguous (R2 self)
 
torch.Tensor copy_ (R2 self, torch.Tensor arg0, bool arg1)
 
bool defined (R2 self)
 
R2 detach (R2 self)
 
torch.Tensor detach_ (R2 self)
 
int dim (R2 self)
 
R2 inv (R2 self)
 
Number|torch.SymInt|torch.SymFloat|torch.SymBool item (R2 self)
 
Scalar norm (R2 self)
 
torch.Tensor requires_grad_ (R2 self, bool arg0)
 
Tensor tensor (R2 self)
 
R2 to (R2 self, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (R2 self)
 
torch.Tensor zero_ (R2 self)
 
R2BaseView base (self)
 
R2BatchView batch (self)
 
torch.device device (self)
 
torch.dtype dtype (self)
 
R2DynamicView dynamic (self)
 
torch.Tensor grad (self)
 
R2IntmdView intmd (self)
 
bool requires_grad (self)
 
tuple[int,...] shape (self)
 
R2StaticView static (self)
 

Static Public Member Functions

R2 dynamic_linspace (R2 start, R2 end, int step, int dim=0)
 
R2 dynamic_logspace (R2 start, R2 end, int step, int dim=0, Number|torch.SymInt|torch.SymFloat|torch.SymBool base=10.0)
 
R2 empty (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 empty (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes=(), *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 empty_like (R2 arg0)
 
SR2 fill (float a, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
SR2 fill (Scalar arg0)
 
SR2 fill (float a11, float a22, float a33, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
SR2 fill (Scalar arg0, Scalar arg1, Scalar arg2)
 
SR2 fill (float a11, float a22, float a33, float a23, float a13, float a12, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
SR2 fill (Scalar arg0, Scalar arg1, Scalar arg2, Scalar arg3, Scalar arg4, Scalar arg5)
 
R2 fill (float a11, float a12, float a13, float a21, float a22, float a23, float a31, float a32, float a33, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 fill (Scalar arg0, Scalar arg1, Scalar arg2, Scalar arg3, Scalar arg4, Scalar arg5, Scalar arg6, Scalar arg7, Scalar arg8)
 
R2 full (float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 full (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 full_like (R2 other, float fill_value)
 
R2 intmd_linspace (R2 start, R2 end, int step, int dim=0)
 
R2 intmd_logspace (R2 start, R2 end, int step, int dim=0, Number|torch.SymInt|torch.SymFloat|torch.SymBool base=10.0)
 
R2 ones (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 ones (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes=(), *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 ones_like (R2 arg0)
 
R2 rand (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 rand (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 rand_like (R2 arg0)
 
R2 zeros (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 zeros (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes=(), *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 zeros_like (R2 arg0)
 

Constructor & Destructor Documentation

◆ __init__() [1/21]

None __init__ ( self)

◆ __init__() [2/21]

None __init__ ( self,
torch.Tensor tensor,
int dynamic_dim,
int intmd_dim )

◆ __init__() [3/21]

None __init__ ( self,
Vec other )

◆ __init__() [4/21]

None __init__ ( self,
Rot other )

◆ __init__() [5/21]

None __init__ ( self,
WR2 other )

◆ __init__() [6/21]

None __init__ ( self,
R2 other )

◆ __init__() [7/21]

None __init__ ( self,
SR2 other )

◆ __init__() [8/21]

None __init__ ( self,
R3 other )

◆ __init__() [9/21]

None __init__ ( self,
SFR3 other )

◆ __init__() [10/21]

None __init__ ( self,
R4 other )

◆ __init__() [11/21]

None __init__ ( self,
SFFR4 other )

◆ __init__() [12/21]

None __init__ ( self,
WFFR4 other )

◆ __init__() [13/21]

None __init__ ( self,
SSR4 other )

◆ __init__() [14/21]

None __init__ ( self,
SWR4 other )

◆ __init__() [15/21]

None __init__ ( self,
WSR4 other )

◆ __init__() [16/21]

None __init__ ( self,
WWR4 other )

◆ __init__() [17/21]

None __init__ ( self,
Quaternion other )

◆ __init__() [18/21]

None __init__ ( self,
MillerIndex other )

◆ __init__() [19/21]

None __init__ ( self,
Tensor other )

◆ __init__() [20/21]

None __init__ ( self,
Scalar other )

◆ __init__() [21/21]

None __init__ ( self,
torch.Tensor tensor,
int intmd_dim = 0 )

Member Function Documentation

◆ __add__() [1/3]

R2 __add__ ( R2 self,
float arg0 )

◆ __add__() [2/3]

R2 __add__ ( R2 self,
R2 arg0 )

◆ __add__() [3/3]

R2 __add__ ( R2 self,
Scalar arg0 )

◆ __iadd__()

R2 __iadd__ ( R2 self,
float arg0 )

◆ __imul__()

R2 __imul__ ( R2 self,
float arg0 )

◆ __isub__()

R2 __isub__ ( R2 self,
float arg0 )

◆ __itruediv__()

R2 __itruediv__ ( R2 self,
float arg0 )

◆ __mul__() [1/4]

R2 __mul__ ( R2 self,
float arg0 )

◆ __mul__() [2/4]

R2 __mul__ ( R2 self,
R2 arg0 )

◆ __mul__() [3/4]

R2 __mul__ ( R2 self,
Scalar arg0 )

◆ __mul__() [4/4]

Vec __mul__ ( R2 self,
Vec arg0 )

◆ __neg__()

R2 __neg__ ( R2 self)

◆ __pow__()

R2 __pow__ ( R2 self,
float arg0 )

◆ __radd__() [1/2]

R2 __radd__ ( R2 self,
float arg0 )

◆ __radd__() [2/2]

R2 __radd__ ( R2 self,
Scalar arg0 )

◆ __repr__()

str __repr__ ( R2 self)

◆ __rmul__() [1/2]

R2 __rmul__ ( R2 self,
float arg0 )

◆ __rmul__() [2/2]

R2 __rmul__ ( R2 self,
Scalar arg0 )

◆ __rsub__() [1/2]

R2 __rsub__ ( R2 self,
float arg0 )

◆ __rsub__() [2/2]

R2 __rsub__ ( R2 self,
Scalar arg0 )

◆ __rtruediv__() [1/2]

R2 __rtruediv__ ( R2 self,
float arg0 )

◆ __rtruediv__() [2/2]

R2 __rtruediv__ ( R2 self,
Scalar arg0 )

◆ __str__()

str __str__ ( R2 self)

◆ __sub__() [1/3]

R2 __sub__ ( R2 self,
float arg0 )

◆ __sub__() [2/3]

R2 __sub__ ( R2 self,
R2 arg0 )

◆ __sub__() [3/3]

R2 __sub__ ( R2 self,
Scalar arg0 )

◆ __truediv__() [1/2]

R2 __truediv__ ( R2 self,
float arg0 )

◆ __truediv__() [2/2]

R2 __truediv__ ( R2 self,
Scalar arg0 )

◆ base()

R2BaseView base ( self)

◆ batch()

R2BatchView batch ( self)

◆ clone()

R2 clone ( R2 self)

◆ contiguous()

R2 contiguous ( R2 self)

◆ copy_()

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

◆ defined()

bool defined ( R2 self)

◆ detach()

R2 detach ( R2 self)

◆ detach_()

torch.Tensor detach_ ( R2 self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( R2 self)

◆ dtype()

torch.dtype dtype ( self)

◆ dynamic()

R2DynamicView dynamic ( self)

◆ dynamic_linspace()

R2 dynamic_linspace ( R2 start,
R2 end,
int step,
int dim = 0 )
static

◆ dynamic_logspace()

R2 dynamic_logspace ( R2 start,
R2 end,
int step,
int dim = 0,
Number | torch.SymInt | torch.SymFloat | torch.SymBool base = 10.0 )
static

◆ empty() [1/2]

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

◆ empty() [2/2]

R2 empty ( tuple[int, ...] dynamic_sizes,
tuple[int, ...] intmd_sizes = (),
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ empty_like()

R2 empty_like ( R2 arg0)
static

◆ fill() [1/8]

SR2 fill ( float a,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ fill() [2/8]

R2 fill ( float a11,
float a12,
float a13,
float a21,
float a22,
float a23,
float a31,
float a32,
float a33,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ fill() [3/8]

SR2 fill ( float a11,
float a22,
float a33,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ fill() [4/8]

SR2 fill ( float a11,
float a22,
float a33,
float a23,
float a13,
float a12,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ fill() [5/8]

SR2 fill ( Scalar arg0)
static

◆ fill() [6/8]

SR2 fill ( Scalar arg0,
Scalar arg1,
Scalar arg2 )
static

◆ fill() [7/8]

SR2 fill ( Scalar arg0,
Scalar arg1,
Scalar arg2,
Scalar arg3,
Scalar arg4,
Scalar arg5 )
static

◆ fill() [8/8]

R2 fill ( Scalar arg0,
Scalar arg1,
Scalar arg2,
Scalar arg3,
Scalar arg4,
Scalar arg5,
Scalar arg6,
Scalar arg7,
Scalar arg8 )
static

◆ full() [1/2]

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

◆ full() [2/2]

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

◆ full_like()

R2 full_like ( R2 other,
float fill_value )
static

◆ grad()

torch.Tensor grad ( self)

◆ intmd()

R2IntmdView intmd ( self)

◆ intmd_linspace()

R2 intmd_linspace ( R2 start,
R2 end,
int step,
int dim = 0 )
static

◆ intmd_logspace()

R2 intmd_logspace ( R2 start,
R2 end,
int step,
int dim = 0,
Number | torch.SymInt | torch.SymFloat | torch.SymBool base = 10.0 )
static

◆ inv()

R2 inv ( R2 self)

◆ item()

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

◆ norm()

Scalar norm ( R2 self)

◆ ones() [1/2]

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

◆ ones() [2/2]

R2 ones ( tuple[int, ...] dynamic_sizes,
tuple[int, ...] intmd_sizes = (),
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ ones_like()

R2 ones_like ( R2 arg0)
static

◆ rand() [1/2]

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

◆ rand() [2/2]

R2 rand ( tuple[int, ...] dynamic_sizes,
tuple[int, ...] intmd_sizes,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ rand_like()

R2 rand_like ( R2 arg0)
static

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

torch.Tensor requires_grad_ ( R2 self,
bool arg0 )

◆ shape()

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

◆ static()

R2StaticView static ( self)

◆ tensor()

Tensor tensor ( R2 self)

◆ to()

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

◆ torch()

torch.Tensor torch ( R2 self)

◆ zero_()

torch.Tensor zero_ ( R2 self)

◆ zeros() [1/2]

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

◆ zeros() [2/2]

R2 zeros ( tuple[int, ...] dynamic_sizes,
tuple[int, ...] intmd_sizes = (),
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ zeros_like()

R2 zeros_like ( R2 arg0)
static