|
NEML2 2.1.0
|
Public Member Functions | |
| neml2.Quaternion | __getitem__ (self, list[typing.Any] arg0) |
| neml2.Quaternion | __getitem__ (self, typing.Any 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, neml2.Tensor arg1) |
| None | __setitem__ (self, typing.Any arg0, neml2.Tensor arg1) |
| int | dim (self) |
| neml2.Quaternion | expand (self, tuple[int,...] arg0, tuple[int,...] arg1) |
| neml2.Quaternion | expand_as (self, neml2.Tensor arg0) |
| neml2.Quaternion | flatten (self) |
| neml2.Quaternion | movedim (self, int arg0, int arg1) |
| neml2.Quaternion | reshape (self, tuple[int,...] arg0, tuple[int,...] arg1) |
| int | size (self, int arg0) |
| neml2.Quaternion | slice (self, int arg0, typing.Any arg1) |
| neml2.Quaternion | squeeze (self, int arg0) |
| neml2.Quaternion | transpose (self, int arg0, int arg1) |
| neml2.Quaternion | unsqueeze (self, int dim, int n=1) |
| tuple | shape (self) |
| neml2.Quaternion __getitem__ | ( | self, | |
| list[typing.Any] | arg0 ) |
| neml2.Quaternion __getitem__ | ( | self, | |
| typing.Any | arg0 ) |
| None __setitem__ | ( | self, | |
| list[typing.Any] | arg0, | ||
| neml2.Tensor | arg1 ) |
| None __setitem__ | ( | self, | |
| list[typing.Any] | arg0, | ||
| torch.Tensor | arg1 ) |
| None __setitem__ | ( | self, | |
| typing.Any | arg0, | ||
| neml2.Tensor | arg1 ) |
| None __setitem__ | ( | self, | |
| typing.Any | arg0, | ||
| torch.Tensor | arg1 ) |
| int dim | ( | self | ) |
| neml2.Quaternion expand | ( | self, | |
| tuple[int, ...] | arg0, | ||
| tuple[int, ...] | arg1 ) |
| neml2.Quaternion expand_as | ( | self, | |
| neml2.Tensor | arg0 ) |
| neml2.Quaternion flatten | ( | self | ) |
| neml2.Quaternion movedim | ( | self, | |
| int | arg0, | ||
| int | arg1 ) |
| neml2.Quaternion reshape | ( | self, | |
| tuple[int, ...] | arg0, | ||
| tuple[int, ...] | arg1 ) |
| tuple shape | ( | self | ) |
| int size | ( | self, | |
| int | arg0 ) |
| neml2.Quaternion slice | ( | self, | |
| int | arg0, | ||
| typing.Any | arg1 ) |
| neml2.Quaternion squeeze | ( | self, | |
| int | arg0 ) |
| neml2.Quaternion transpose | ( | self, | |
| int | arg0, | ||
| int | arg1 ) |
| neml2.Quaternion unsqueeze | ( | self, | |
| int | dim, | ||
| int | n = 1 ) |