NEML2 1.4.0
Loading...
Searching...
No Matches
Model Member List

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