|
NEML2 2.0.0
|
Public Member Functions | |
| None | __init__ (self, Rot arg0) |
| int | dim (self) |
| Tensor | expand (self, tuple[int,...] arg0, tuple[int,...] arg1) |
| Tensor | expand_as (self, Tensor arg0) |
| Tensor | flatten (self) |
| Tensor | reshape (self, tuple[int,...] arg0, tuple[int,...] arg1) |
| int | size (self, int arg0) |
| tuple | shape (self) |
| None __init__ | ( | self, | |
| Rot | arg0 ) |
| int dim | ( | self | ) |
| Tensor expand | ( | self, | |
| tuple[int, ...] | arg0, | ||
| tuple[int, ...] | arg1 ) |
| Tensor flatten | ( | self | ) |
| Tensor reshape | ( | self, | |
| tuple[int, ...] | arg0, | ||
| tuple[int, ...] | arg1 ) |
| tuple shape | ( | self | ) |
| int size | ( | self, | |
| int | arg0 ) |