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

Public Member Functions

Tensor __add__ (Tensor self, Tensor arg0)
 
Tensor __add__ (Tensor self, Scalar arg0)
 
Tensor __add__ (Tensor self, float arg0)
 
Tensor __iadd__ (Tensor self, float arg0)
 
Tensor __imul__ (Tensor 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 dynamic_dim, int intmd_dim=0)
 
Tensor __isub__ (Tensor self, float arg0)
 
Tensor __itruediv__ (Tensor self, float arg0)
 
Tensor __mul__ (Tensor self, Tensor arg0)
 
Tensor __mul__ (Tensor self, Scalar arg0)
 
Tensor __mul__ (Tensor self, float arg0)
 
Tensor __neg__ (Tensor self)
 
Tensor __pow__ (Tensor self, float arg0)
 
Tensor __pow__ (Tensor self, Tensor arg0)
 
Tensor __radd__ (Tensor self, Scalar arg0)
 
Tensor __radd__ (Tensor self, float arg0)
 
str __repr__ (Tensor self)
 
Tensor __rmul__ (Tensor self, Scalar arg0)
 
Tensor __rmul__ (Tensor self, float arg0)
 
Tensor __rpow__ (Tensor self, float arg0)
 
Tensor __rsub__ (Tensor self, Scalar arg0)
 
Tensor __rsub__ (Tensor self, float arg0)
 
Tensor __rtruediv__ (Tensor self, Scalar arg0)
 
Tensor __rtruediv__ (Tensor self, float arg0)
 
str __str__ (Tensor self)
 
Tensor __sub__ (Tensor self, Tensor arg0)
 
Tensor __sub__ (Tensor self, Scalar arg0)
 
Tensor __sub__ (Tensor self, float arg0)
 
Tensor __truediv__ (Tensor self, Tensor arg0)
 
Tensor __truediv__ (Tensor self, Scalar arg0)
 
Tensor __truediv__ (Tensor self, float arg0)
 
Tensor clone (Tensor self)
 
Tensor contiguous (Tensor self)
 
torch.Tensor copy_ (Tensor self, torch.Tensor arg0, bool arg1)
 
bool defined (Tensor self)
 
Tensor detach (Tensor self)
 
torch.Tensor detach_ (Tensor self)
 
int dim (Tensor self)
 
Tensor inv (Tensor self)
 
Number|torch.SymInt|torch.SymFloat|torch.SymBool item (Tensor self)
 
Scalar norm (Tensor self)
 
torch.Tensor requires_grad_ (Tensor self, bool arg0)
 
Tensor tensor (Tensor self)
 
Tensor to (Tensor self, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (Tensor self)
 
torch.Tensor zero_ (Tensor self)
 
TensorBaseView base (self)
 
TensorBatchView batch (self)
 
torch.device device (self)
 
torch.dtype dtype (self)
 
TensorDynamicView dynamic (self)
 
torch.Tensor grad (self)
 
TensorIntmdView intmd (self)
 
bool requires_grad (self)
 
tuple[int,...] shape (self)
 
TensorStaticView static (self)
 

Static Public Member Functions

Tensor base_linspace (Tensor start, Tensor end, int step, int dim=0)
 
Tensor base_logspace (Tensor start, Tensor end, int step, int dim=0, Number|torch.SymInt|torch.SymFloat|torch.SymBool base=10.0)
 
Tensor dynamic_linspace (Tensor start, Tensor end, int step, int dim=0)
 
Tensor dynamic_logspace (Tensor start, Tensor end, int step, int dim=0, Number|torch.SymInt|torch.SymFloat|torch.SymBool base=10.0)
 
Tensor empty (tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor empty (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor empty_like (Tensor arg0)
 
Tensor full (tuple[int,...] base_sizes, float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor full (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, tuple[int,...] base_sizes, float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor full_like (Tensor other, float fill_value)
 
Tensor identity (int n, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor intmd_linspace (Tensor start, Tensor end, int step, int dim=0)
 
Tensor intmd_logspace (Tensor start, Tensor end, int step, int dim=0, Number|torch.SymInt|torch.SymFloat|torch.SymBool base=10.0)
 
Tensor ones (tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor ones (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor ones_like (Tensor arg0)
 
Tensor rand (tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor rand (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor rand_like (Tensor arg0)
 
Tensor zeros (tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor zeros (tuple[int,...] dynamic_sizes, tuple[int,...] intmd_sizes, tuple[int,...] base_sizes, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor zeros_like (Tensor 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 dynamic_dim,
int intmd_dim = 0 )

Member Function Documentation

◆ __add__() [1/3]

Tensor __add__ ( Tensor self,
float arg0 )

◆ __add__() [2/3]

Tensor __add__ ( Tensor self,
Scalar arg0 )

◆ __add__() [3/3]

Tensor __add__ ( Tensor self,
Tensor arg0 )

◆ __iadd__()

Tensor __iadd__ ( Tensor self,
float arg0 )

◆ __imul__()

Tensor __imul__ ( Tensor self,
float arg0 )

◆ __isub__()

Tensor __isub__ ( Tensor self,
float arg0 )

◆ __itruediv__()

Tensor __itruediv__ ( Tensor self,
float arg0 )

◆ __mul__() [1/3]

Tensor __mul__ ( Tensor self,
float arg0 )

◆ __mul__() [2/3]

Tensor __mul__ ( Tensor self,
Scalar arg0 )

◆ __mul__() [3/3]

Tensor __mul__ ( Tensor self,
Tensor arg0 )

◆ __neg__()

Tensor __neg__ ( Tensor self)

◆ __pow__() [1/2]

Tensor __pow__ ( Tensor self,
float arg0 )

◆ __pow__() [2/2]

Tensor __pow__ ( Tensor self,
Tensor arg0 )

◆ __radd__() [1/2]

Tensor __radd__ ( Tensor self,
float arg0 )

◆ __radd__() [2/2]

Tensor __radd__ ( Tensor self,
Scalar arg0 )

◆ __repr__()

str __repr__ ( Tensor self)

◆ __rmul__() [1/2]

Tensor __rmul__ ( Tensor self,
float arg0 )

◆ __rmul__() [2/2]

Tensor __rmul__ ( Tensor self,
Scalar arg0 )

◆ __rpow__()

Tensor __rpow__ ( Tensor self,
float arg0 )

◆ __rsub__() [1/2]

Tensor __rsub__ ( Tensor self,
float arg0 )

◆ __rsub__() [2/2]

Tensor __rsub__ ( Tensor self,
Scalar arg0 )

◆ __rtruediv__() [1/2]

Tensor __rtruediv__ ( Tensor self,
float arg0 )

◆ __rtruediv__() [2/2]

Tensor __rtruediv__ ( Tensor self,
Scalar arg0 )

◆ __str__()

str __str__ ( Tensor self)

◆ __sub__() [1/3]

Tensor __sub__ ( Tensor self,
float arg0 )

◆ __sub__() [2/3]

Tensor __sub__ ( Tensor self,
Scalar arg0 )

◆ __sub__() [3/3]

Tensor __sub__ ( Tensor self,
Tensor arg0 )

◆ __truediv__() [1/3]

Tensor __truediv__ ( Tensor self,
float arg0 )

◆ __truediv__() [2/3]

Tensor __truediv__ ( Tensor self,
Scalar arg0 )

◆ __truediv__() [3/3]

Tensor __truediv__ ( Tensor self,
Tensor arg0 )

◆ base()

TensorBaseView base ( self)

◆ base_linspace()

Tensor base_linspace ( Tensor start,
Tensor end,
int step,
int dim = 0 )
static

◆ base_logspace()

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

◆ batch()

TensorBatchView batch ( self)

◆ clone()

Tensor clone ( Tensor self)

◆ contiguous()

Tensor contiguous ( Tensor self)

◆ copy_()

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

◆ defined()

bool defined ( Tensor self)

◆ detach()

Tensor detach ( Tensor self)

◆ detach_()

torch.Tensor detach_ ( Tensor self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( Tensor self)

◆ dtype()

torch.dtype dtype ( self)

◆ dynamic()

TensorDynamicView dynamic ( self)

◆ dynamic_linspace()

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

◆ dynamic_logspace()

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

◆ empty() [1/2]

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

◆ empty() [2/2]

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

◆ empty_like()

Tensor empty_like ( Tensor arg0)
static

◆ full() [1/2]

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

◆ full() [2/2]

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

◆ full_like()

Tensor full_like ( Tensor other,
float fill_value )
static

◆ grad()

torch.Tensor grad ( self)

◆ identity()

Tensor identity ( int n,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ intmd()

TensorIntmdView intmd ( self)

◆ intmd_linspace()

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

◆ intmd_logspace()

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

◆ inv()

Tensor inv ( Tensor self)

◆ item()

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

◆ norm()

Scalar norm ( Tensor self)

◆ ones() [1/2]

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

◆ ones() [2/2]

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

◆ ones_like()

Tensor ones_like ( Tensor arg0)
static

◆ rand() [1/2]

Tensor rand ( tuple[int, ...] base_sizes,
* ,
torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ rand() [2/2]

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

◆ rand_like()

Tensor rand_like ( Tensor arg0)
static

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

torch.Tensor requires_grad_ ( Tensor self,
bool arg0 )

◆ shape()

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

◆ static()

TensorStaticView static ( self)

◆ tensor()

Tensor tensor ( Tensor self)

◆ to()

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

◆ torch()

torch.Tensor torch ( Tensor self)

◆ zero_()

torch.Tensor zero_ ( Tensor self)

◆ zeros() [1/2]

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

◆ zeros() [2/2]

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

◆ zeros_like()

Tensor zeros_like ( Tensor arg0)
static