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

Public Member Functions

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

Constructor & Destructor Documentation

◆ __init__()

None __init__ ( self,
WR2 arg0 )

Member Function Documentation

◆ __getitem__() [1/2]

WR2 __getitem__ ( self,
[typing.Any, ...] arg0 )

◆ __getitem__() [2/2]

WR2 __getitem__ ( self,
typing.Any arg0 )

◆ __setitem__() [1/4]

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

◆ __setitem__() [2/4]

None __setitem__ ( self,
[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()

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

◆ expand_copy()

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

◆ shape()

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

◆ transpose()

WR2 transpose ( self,
int arg0,
int arg1 )

◆ unsqueeze()

WR2 unsqueeze ( self,
int arg0 )