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

Public Member Functions

None __init__ (self, SWR4 arg0)
 
int dim (self)
 
SWR4 expand (self, tuple[int,...] arg0, tuple[int,...] arg1)
 
SWR4 expand_as (self, Tensor arg0)
 
SWR4 flatten (self)
 
SWR4 reshape (self, tuple[int,...] arg0, tuple[int,...] arg1)
 
int size (self, int arg0)
 
tuple shape (self)
 

Constructor & Destructor Documentation

◆ __init__()

None __init__ ( self,
SWR4 arg0 )

Member Function Documentation

◆ dim()

int dim ( self)

◆ expand()

SWR4 expand ( self,
tuple[int, ...] arg0,
tuple[int, ...] arg1 )

◆ expand_as()

SWR4 expand_as ( self,
Tensor arg0 )

◆ flatten()

SWR4 flatten ( self)

◆ reshape()

SWR4 reshape ( self,
tuple[int, ...] arg0,
tuple[int, ...] arg1 )

◆ shape()

tuple shape ( self)

◆ size()

int size ( self,
int arg0 )