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

Public Member Functions

Tensor __getitem__ (self, list[typing.Any] arg0)
 
Tensor __getitem__ (self, typing.Any arg0)
 
None __init__ (self, Tensor arg0)
 
None __setitem__ (self, list[typing.Any] arg0, torch.Tensor arg1)
 
None __setitem__ (self, typing.Any arg0, torch.Tensor arg1)
 
None __setitem__ (self, list[typing.Any] arg0, Tensor arg1)
 
None __setitem__ (self, typing.Any arg0, Tensor arg1)
 
int dim (self)
 
Tensor expand (self, tuple[int,...] arg0)
 
Tensor expand_copy (self, tuple[int,...] arg0)
 
Tensor reshape (self, tuple[int,...] arg0)
 
Tensor transpose (self, int arg0, int arg1)
 
Tensor unsqueeze (self, int arg0)
 
tuple shape (self)
 

Constructor & Destructor Documentation

◆ __init__()

None __init__ ( self,
Tensor arg0 )

Member Function Documentation

◆ __getitem__() [1/2]

Tensor __getitem__ ( self,
list[typing.Any] arg0 )

◆ __getitem__() [2/2]

Tensor __getitem__ ( self,
typing.Any arg0 )

◆ __setitem__() [1/4]

None __setitem__ ( self,
list[typing.Any] arg0,
Tensor arg1 )

◆ __setitem__() [2/4]

None __setitem__ ( self,
list[typing.Any] arg0,
torch.Tensor arg1 )

◆ __setitem__() [3/4]

None __setitem__ ( self,
typing.Any arg0,
Tensor arg1 )

◆ __setitem__() [4/4]

None __setitem__ ( self,
typing.Any arg0,
torch.Tensor arg1 )

◆ dim()

int dim ( self)

◆ expand()

Tensor expand ( self,
tuple[int, ...] arg0 )

◆ expand_copy()

Tensor expand_copy ( self,
tuple[int, ...] arg0 )

◆ reshape()

Tensor reshape ( self,
tuple[int, ...] arg0 )

◆ shape()

tuple shape ( self)

◆ transpose()

Tensor transpose ( self,
int arg0,
int arg1 )

◆ unsqueeze()

Tensor unsqueeze ( self,
int arg0 )