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