|
NEML2 2.1.0
|
Layout of variables along an equation-system axis
Classes | |
| class | IStructure |
Public Member Functions | |
| bool | __eq__ (self, AxisLayout arg0) |
| None | __init__ (self) |
| None | __init__ (self, list[list[str]] vars, list[tuple[int,...]] intmd_shapes, list[tuple[int,...]] base_shapes, list[AxisLayout.IStructure] istrs) |
| tuple[int,...] | base_sizes (self, int i) |
| AxisLayout | group (self, int i) |
| tuple[int, int] | group_offsets (self, int i) |
| tuple[int,...] | intmd_sizes (self, int i) |
| AxisLayout.IStructure | istr (self, int i=0) |
| int | ngroup (self) |
| int | nvar (self) |
| list[int] | storage_sizes (self, bool include_intmd) |
| str | var (self, int i) |
| list[str] | vars (self) |
| AxisLayout | view (self) |
| bool | is_view (self) |
Static Public Attributes | |
| typing | DENSE .ClassVar[AxisLayout.IStructure] |
| None __init__ | ( | self | ) |
| None __init__ | ( | self, | |
| list[list[str]] | vars, | ||
| list[tuple[int, ...]] | intmd_shapes, | ||
| list[tuple[int, ...]] | base_shapes, | ||
| list[AxisLayout.IStructure] | istrs ) |
Construct an AxisLayout. :param vars: List of variable groups; each group is a list of variable names :param intmd_shapes: Intermediate shape for each variable (flattened across all groups) :param base_shapes: Base shape for each variable (flattened across all groups) :param istrs: IStructure for each variable group
| bool __eq__ | ( | self, | |
| AxisLayout | arg0 ) |
| tuple[int, ...] base_sizes | ( | self, | |
| int | i ) |
Base shape of variable i
| AxisLayout group | ( | self, | |
| int | i ) |
Contiguous view of variable group i
| tuple[int, int] group_offsets | ( | self, | |
| int | i ) |
Starting and ending offsets (as a tuple) of variable group i
| tuple[int, ...] intmd_sizes | ( | self, | |
| int | i ) |
Intermediate shape of variable i
| bool is_view | ( | self | ) |
Whether this is a view into a parent layout
| AxisLayout.IStructure istr | ( | self, | |
| int | i = 0 ) |
IStructure of variable group i
| int ngroup | ( | self | ) |
Number of variable groups
| int nvar | ( | self | ) |
Number of variables
| list[int] storage_sizes | ( | self, | |
| bool | include_intmd ) |
Storage size of each variable; pass include_intmd=True to include intermediate dimensions
| str var | ( | self, | |
| int | i ) |
Name of variable i
| list[str] vars | ( | self | ) |
List of variable names
| AxisLayout view | ( | self | ) |
Contiguous view of the entire layout
|
static |