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

Public Member Functions

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

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

◆ __init__() [4/5]

None __init__ ( self,
torch.Tensor arg0 )

◆ __init__() [5/5]

None __init__ ( self,
Rot arg0 )

Member Function Documentation

◆ __add__() [1/3]

Vec __add__ ( self,
float arg0 )

◆ __add__() [2/3]

Vec __add__ ( self,
Scalar arg0 )

◆ __add__() [3/3]

Vec __add__ ( self,
Vec arg0 )

◆ __call__()

Scalar __call__ ( self,
int arg0 )

◆ __mul__() [1/2]

Vec __mul__ ( self,
float arg0 )

◆ __mul__() [2/2]

Vec __mul__ ( self,
Scalar arg0 )

◆ __neg__()

Vec __neg__ ( self)

◆ __pow__() [1/2]

Vec __pow__ ( self,
float arg0 )

◆ __pow__() [2/2]

Vec __pow__ ( self,
Scalar arg0 )

◆ __radd__()

Vec __radd__ ( self,
float arg0 )

◆ __repr__()

str __repr__ ( self)

◆ __rmul__()

Vec __rmul__ ( self,
float arg0 )

◆ __rpow__()

Tensor __rpow__ ( self,
float arg0 )

◆ __rsub__()

Vec __rsub__ ( self,
float arg0 )

◆ __rtruediv__()

Vec __rtruediv__ ( self,
float arg0 )

◆ __str__()

str __str__ ( self)

◆ __sub__() [1/3]

Vec __sub__ ( self,
float arg0 )

◆ __sub__() [2/3]

Vec __sub__ ( self,
Scalar arg0 )

◆ __sub__() [3/3]

Vec __sub__ ( self,
Vec arg0 )

◆ __truediv__() [1/3]

Vec __truediv__ ( self,
float arg0 )

◆ __truediv__() [2/3]

Vec __truediv__ ( self,
Scalar arg0 )

◆ __truediv__() [3/3]

Vec __truediv__ ( self,
Vec arg0 )

◆ base()

VecBaseView base ( self)

◆ batch()

VecBatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

Vec clone ( self)

◆ copy_()

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

◆ cross() [1/3]

Vec cross ( self,
Rot arg0 )

◆ cross() [2/3]

Vec cross ( self,
Vec arg0 )

◆ cross() [3/3]

Vec cross ( self,
WR2 arg0 )

◆ defined()

bool defined ( self)

◆ detach()

Vec detach ( self)

◆ detach_()

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

R2 drotate ( self,
Rot arg0 )

◆ dtype()

torch.dtype dtype ( self)

◆ empty() [1/2]

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

◆ empty() [2/2]

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

◆ empty_like()

Vec empty_like ( Vec arg0)
static

◆ fill() [1/2]

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

◆ fill() [2/2]

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

◆ full() [1/2]

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

◆ full() [2/2]

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

◆ full_like()

Vec full_like ( Vec 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()

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

◆ logspace()

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

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

◆ ones() [2/2]

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

◆ ones_like()

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

Vec rotate ( self,
Rot arg0 )

◆ shape()

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

◆ tensor()

Tensor tensor ( self)

◆ to()

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

◆ transform()

Vec transform ( self,
R2 arg0 )

◆ zero_()

torch.Tensor zero_ ( self)

◆ zeros() [1/2]

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

◆ zeros() [2/2]

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

◆ zeros_like()

Vec zeros_like ( Vec arg0)
static