NEML2 1.4.0
|
This is the complete list of members for Model, including all inherited members.
dependency(self) | Model | |
get_parameter(self, str arg0) | Model | |
input_axis(self) | Model | |
input_type(self, tensors.LabeledAxisAccessor variable) | Model | |
is_AD_enabled(self) | Model | |
name(self) | Model | |
named_buffers(self) | Model | |
named_parameters(self) | Model | |
named_submodels(self) | Model | |
output_axis(self) | Model | |
output_type(self, tensors.LabeledAxisAccessor variable) | Model | |
reinit(self, tuple[int,...] batch_shape=(), int deriv_order=0, torch.device device=..., torch.dtype dtype=...) | Model | |
set_parameter(self, str arg0, tensors.Tensor arg1) | Model | |
set_parameters(self, dict[str, tensors.Tensor] arg0) | Model | |
type(self) | Model | |
value(self, typing.Any arg0) | Model | |
value_and_dvalue(self, typing.Any arg0) | Model |