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

Public Member Functions

Tensor __add__ (self, float arg0)
 
Tensor __add__ (self, Scalar arg0)
 
Tensor __add__ (self, Tensor arg0)
 
None __init__ (self)
 
None __init__ (self, torch.Tensor arg0, int arg1)
 
None __init__ (self, Tensor arg0)
 
None __init__ (self, Vec arg0)
 
None __init__ (self, Rot arg0)
 
None __init__ (self, WR2 arg0)
 
None __init__ (self, R2 arg0)
 
None __init__ (self, Scalar arg0)
 
None __init__ (self, SR2 arg0)
 
None __init__ (self, R3 arg0)
 
None __init__ (self, SFR3 arg0)
 
None __init__ (self, R4 arg0)
 
None __init__ (self, SSR4 arg0)
 
None __init__ (self, SFFR4 arg0)
 
None __init__ (self, SWR4 arg0)
 
None __init__ (self, WSR4 arg0)
 
None __init__ (self, WWR4 arg0)
 
None __init__ (self, R5 arg0)
 
None __init__ (self, SSFR5 arg0)
 
None __init__ (self, Quaternion arg0)
 
None __init__ (self, MillerIndex arg0)
 
Tensor __mul__ (self, float arg0)
 
Tensor __mul__ (self, Scalar arg0)
 
Tensor __neg__ (self)
 
Tensor __pow__ (self, float arg0)
 
Tensor __pow__ (self, Scalar arg0)
 
Tensor __pow__ (self, Tensor arg0)
 
Tensor __radd__ (self, float arg0)
 
str __repr__ (self)
 
Tensor __rmul__ (self, float arg0)
 
Tensor __rpow__ (self, float arg0)
 
Tensor __rsub__ (self, float arg0)
 
Tensor __rtruediv__ (self, float arg0)
 
str __str__ (self)
 
Tensor __sub__ (self, float arg0)
 
Tensor __sub__ (self, Scalar arg0)
 
Tensor __sub__ (self, Tensor arg0)
 
Tensor __truediv__ (self, float arg0)
 
Tensor __truediv__ (self, Scalar arg0)
 
Tensor __truediv__ (self, Tensor arg0)
 
bool batched (self)
 
Tensor clone (self)
 
torch.Tensor copy_ (self, torch.Tensor arg0, bool arg1)
 
bool defined (self)
 
Tensor detach (self)
 
torch.Tensor detach_ (self)
 
int dim (self)
 
torch.Tensor requires_grad_ (self, bool arg0)
 
Tensor to (self, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (self)
 
torch.Tensor zero_ (self)
 
TensorBaseView base (self)
 
TensorBatchView 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

Tensor empty (tuple[int,...] base_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor empty (tuple[int,...] batch_shape, tuple[int,...] base_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor empty_like (Tensor arg0)
 
Tensor full (tuple[int,...] base_shape, float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor full (tuple[int,...] batch_shape, tuple[int,...] base_shape, float fill_value, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor full_like (Tensor arg0, float arg1)
 
Tensor identity (int n, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor identity (tuple[int,...] batch_shape, int n, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor linspace (Tensor start, Tensor end, int nstep, int dim=0, int batch_dim=-1)
 
Tensor logspace (Tensor start, Tensor end, int nstep, int dim=0, int batch_dim=-1, float base=10.0)
 
Tensor ones (tuple[int,...] base_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor ones (tuple[int,...] batch_shape, tuple[int,...] base_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor ones_like (Tensor arg0)
 
Tensor zeros (tuple[int,...] base_shape, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
Tensor zeros (tuple[int,...] batch_shape, tuple[int,...] base_shape, *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 arg0,
int arg1 )

◆ __init__() [3/21]

None __init__ ( self,
Tensor arg0 )

◆ __init__() [4/21]

None __init__ ( self,
Vec arg0 )

◆ __init__() [5/21]

None __init__ ( self,
Rot arg0 )

◆ __init__() [6/21]

None __init__ ( self,
WR2 arg0 )

◆ __init__() [7/21]

None __init__ ( self,
R2 arg0 )

◆ __init__() [8/21]

None __init__ ( self,
Scalar arg0 )

◆ __init__() [9/21]

None __init__ ( self,
SR2 arg0 )

◆ __init__() [10/21]

None __init__ ( self,
R3 arg0 )

◆ __init__() [11/21]

None __init__ ( self,
SFR3 arg0 )

◆ __init__() [12/21]

None __init__ ( self,
R4 arg0 )

◆ __init__() [13/21]

None __init__ ( self,
SSR4 arg0 )

◆ __init__() [14/21]

None __init__ ( self,
SFFR4 arg0 )

◆ __init__() [15/21]

None __init__ ( self,
SWR4 arg0 )

◆ __init__() [16/21]

None __init__ ( self,
WSR4 arg0 )

◆ __init__() [17/21]

None __init__ ( self,
WWR4 arg0 )

◆ __init__() [18/21]

None __init__ ( self,
R5 arg0 )

◆ __init__() [19/21]

None __init__ ( self,
SSFR5 arg0 )

◆ __init__() [20/21]

None __init__ ( self,
Quaternion arg0 )

◆ __init__() [21/21]

None __init__ ( self,
MillerIndex arg0 )

Member Function Documentation

◆ __add__() [1/3]

Tensor __add__ ( self,
float arg0 )

◆ __add__() [2/3]

Tensor __add__ ( self,
Scalar arg0 )

◆ __add__() [3/3]

Tensor __add__ ( self,
Tensor arg0 )

◆ __mul__() [1/2]

Tensor __mul__ ( self,
float arg0 )

◆ __mul__() [2/2]

Tensor __mul__ ( self,
Scalar arg0 )

◆ __neg__()

Tensor __neg__ ( self)

◆ __pow__() [1/3]

Tensor __pow__ ( self,
float arg0 )

◆ __pow__() [2/3]

Tensor __pow__ ( self,
Scalar arg0 )

◆ __pow__() [3/3]

Tensor __pow__ ( self,
Tensor arg0 )

◆ __radd__()

Tensor __radd__ ( self,
float arg0 )

◆ __repr__()

str __repr__ ( self)

◆ __rmul__()

Tensor __rmul__ ( self,
float arg0 )

◆ __rpow__()

Tensor __rpow__ ( self,
float arg0 )

◆ __rsub__()

Tensor __rsub__ ( self,
float arg0 )

◆ __rtruediv__()

Tensor __rtruediv__ ( self,
float arg0 )

◆ __str__()

str __str__ ( self)

◆ __sub__() [1/3]

Tensor __sub__ ( self,
float arg0 )

◆ __sub__() [2/3]

Tensor __sub__ ( self,
Scalar arg0 )

◆ __sub__() [3/3]

Tensor __sub__ ( self,
Tensor arg0 )

◆ __truediv__() [1/3]

Tensor __truediv__ ( self,
float arg0 )

◆ __truediv__() [2/3]

Tensor __truediv__ ( self,
Scalar arg0 )

◆ __truediv__() [3/3]

Tensor __truediv__ ( self,
Tensor arg0 )

◆ base()

TensorBaseView base ( self)

◆ batch()

TensorBatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

Tensor clone ( self)

◆ copy_()

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

◆ defined()

bool defined ( self)

◆ detach()

Tensor detach ( self)

◆ detach_()

torch.Tensor detach_ ( self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( self)

◆ dtype()

torch.dtype dtype ( self)

◆ empty() [1/2]

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

◆ empty() [2/2]

Tensor empty ( tuple[int, ...] batch_shape,
tuple[int, ...] base_shape,
*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_shape,
float fill_value,
*torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )
static

◆ full() [2/2]

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

◆ full_like()

Tensor full_like ( Tensor arg0,
float arg1 )
static

◆ grad()

torch.Tensor grad ( self)

◆ identity() [1/2]

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

◆ identity() [2/2]

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

◆ linspace()

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

◆ logspace()

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

◆ ones() [1/2]

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

◆ ones() [2/2]

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

◆ ones_like()

Tensor ones_like ( Tensor arg0)
static

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

torch.Tensor requires_grad_ ( self,
bool arg0 )

◆ shape()

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

◆ to()

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

◆ torch()

torch.Tensor torch ( self)

◆ zero_()

torch.Tensor zero_ ( self)

◆ zeros() [1/2]

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

◆ zeros() [2/2]

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

◆ zeros_like()

Tensor zeros_like ( Tensor arg0)
static