NEML2 2.0.0
|
Public Member Functions | |
None | __init__ (self, LabeledAxis arg0, LabeledAxis arg1) |
tensors.Tensor | assemble_by_variable (self, dict arg0) |
dict[LabeledAxisAccessor, dict[LabeledAxisAccessor, tensors.Tensor]] | split_by_subaxis (self, tensors.Tensor arg0) |
dict[LabeledAxisAccessor, dict[LabeledAxisAccessor, tensors.Tensor]] | split_by_variable (self, tensors.Tensor arg0) |
None __init__ | ( | self, | |
LabeledAxis | arg0, | ||
LabeledAxis | arg1 ) |
tensors.Tensor assemble_by_variable | ( | self, | |
dict | arg0 ) |
dict[LabeledAxisAccessor, dict[LabeledAxisAccessor, tensors.Tensor]] split_by_subaxis | ( | self, | |
tensors.Tensor | arg0 ) |
dict[LabeledAxisAccessor, dict[LabeledAxisAccessor, tensors.Tensor]] split_by_variable | ( | self, | |
tensors.Tensor | arg0 ) |