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

Public Member Functions

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

R2 empty (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 empty (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 empty_like (R2 arg0)
 
R2 full (float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 full (tuple[int,...] batch_shape, float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 full_like (R2 arg0, float arg1)
 
R2 linspace (R2 start, R2 end, int nstep, int dim=0, int batch_dim=-1)
 
R2 logspace (R2 start, R2 end, int nstep, int dim=0, int batch_dim=-1, float base=10.0)
 
R2 ones (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 ones (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 ones_like (R2 arg0)
 
R2 zeros (*torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 zeros (tuple[int,...] batch_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
R2 zeros_like (R2 arg0)
 

Constructor & Destructor Documentation

◆ __init__() [1/4]

None __init__ ( self)

◆ __init__() [2/4]

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

◆ __init__() [3/4]

None __init__ ( self,
R2 arg0 )

◆ __init__() [4/4]

None __init__ ( self,
torch.Tensor arg0 )

Member Function Documentation

◆ __add__() [1/3]

R2 __add__ ( self,
float arg0 )

◆ __add__() [2/3]

R2 __add__ ( self,
R2 arg0 )

◆ __add__() [3/3]

R2 __add__ ( self,
Scalar arg0 )

◆ __call__()

Scalar __call__ ( self,
int arg0,
int arg1 )

◆ __mul__() [1/4]

R2 __mul__ ( self,
float arg0 )

◆ __mul__() [2/4]

R2 __mul__ ( self,
R2 arg0 )

◆ __mul__() [3/4]

R2 __mul__ ( self,
Scalar arg0 )

◆ __mul__() [4/4]

Vec __mul__ ( self,
Vec arg0 )

◆ __neg__()

R2 __neg__ ( self)

◆ __pow__() [1/2]

R2 __pow__ ( self,
float arg0 )

◆ __pow__() [2/2]

R2 __pow__ ( self,
Scalar arg0 )

◆ __radd__()

R2 __radd__ ( self,
float arg0 )

◆ __repr__()

str __repr__ ( self)

◆ __rmul__()

R2 __rmul__ ( self,
float arg0 )

◆ __rpow__()

Tensor __rpow__ ( self,
float arg0 )

◆ __rsub__()

R2 __rsub__ ( self,
float arg0 )

◆ __rtruediv__()

R2 __rtruediv__ ( self,
float arg0 )

◆ __str__()

str __str__ ( self)

◆ __sub__() [1/3]

R2 __sub__ ( self,
float arg0 )

◆ __sub__() [2/3]

R2 __sub__ ( self,
R2 arg0 )

◆ __sub__() [3/3]

R2 __sub__ ( self,
Scalar arg0 )

◆ __truediv__() [1/3]

R2 __truediv__ ( self,
float arg0 )

◆ __truediv__() [2/3]

R2 __truediv__ ( self,
R2 arg0 )

◆ __truediv__() [3/3]

R2 __truediv__ ( self,
Scalar arg0 )

◆ base()

R2BaseView base ( self)

◆ batch()

R2BatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

R2 clone ( self)

◆ copy_()

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

◆ defined()

bool defined ( self)

◆ detach()

R2 detach ( self)

◆ detach_()

torch.Tensor detach_ ( self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( self)

◆ drotate()

R3 drotate ( self,
Rot arg0 )

◆ dtype()

torch.dtype dtype ( self)

◆ empty() [1/2]

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

◆ empty() [2/2]

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

◆ empty_like()

R2 empty_like ( R2 arg0)
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, ...] batch_shape,
float fill_value,
*torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ full_like()

R2 full_like ( R2 arg0,
float arg1 )
static

◆ grad()

torch.Tensor grad ( self)

◆ inverse()

R2 inverse ( self)

◆ linspace()

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

◆ logspace()

R2 logspace ( R2 start,
R2 end,
int nstep,
int dim = 0,
int batch_dim = -1,
float base = 10.0 )
static

◆ ones() [1/2]

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

◆ ones() [2/2]

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

◆ ones_like()

R2 ones_like ( R2 arg0)
static

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

torch.Tensor requires_grad_ ( self,
bool arg0 )

◆ rotate()

R2 rotate ( self,
Rot arg0 )

◆ shape()

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

◆ tensor()

Tensor tensor ( self)

◆ to()

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

◆ transpose()

R2 transpose ( self)

◆ zero_()

torch.Tensor zero_ ( self)

◆ zeros() [1/2]

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

◆ zeros() [2/2]

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

◆ zeros_like()

R2 zeros_like ( R2 arg0)
static