NEML2 2.0.0
All Classes Namespaces Functions Variables Pages
WFR4BatchView Class Reference

Public Member Functions

WFR4 __getitem__ (self, list[typing.Any] arg0)
 
WFR4 __getitem__ (self, typing.Any arg0)
 
None __init__ (self, WFR4 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)
 
WFR4 expand (self, tuple[int,...] arg0)
 
WFR4 expand_copy (self, tuple[int,...] arg0)
 
WFR4 reshape (self, tuple[int,...] arg0)
 
WFR4 transpose (self, int arg0, int arg1)
 
WFR4 unsqueeze (self, int arg0)
 
tuple shape (self)
 

Constructor & Destructor Documentation

◆ __init__()

None __init__ ( self,
WFR4 arg0 )

Member Function Documentation

◆ __getitem__() [1/2]

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

◆ __getitem__() [2/2]

WFR4 __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()

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

◆ expand_copy()

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

◆ reshape()

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

◆ shape()

tuple shape ( self)

◆ transpose()

WFR4 transpose ( self,
int arg0,
int arg1 )

◆ unsqueeze()

WFR4 unsqueeze ( self,
int arg0 )