Trait dfdx::nn::tensor_collection::ModuleFields
source · pub trait ModuleFields<M: TensorCollection<E, D>, E: Dtype, D: Device<E>> {
type Options<E2: Dtype, D2: Device<E2>>;
type Output<E2: Dtype, D2: Device<E2>>;
// Required methods
fn visit_fields<V: ModuleVisitor<M, E, D>>(
self,
visitor: &mut V
) -> Result<Self::Options<V::E2, V::D2>, V::Err>;
fn handle_options<E2: Dtype, D2: Device<E2>>(
options: Self::Options<E2, D2>
) -> Option<Self::Output<E2, D2>>;
}
Expand description
A list of a Module’s fields. Used in ModuleVisitor::visit_fields.
Required Associated Types§
Required Methods§
sourcefn visit_fields<V: ModuleVisitor<M, E, D>>(
self,
visitor: &mut V
) -> Result<Self::Options<V::E2, V::D2>, V::Err>
fn visit_fields<V: ModuleVisitor<M, E, D>>( self, visitor: &mut V ) -> Result<Self::Options<V::E2, V::D2>, V::Err>
Calls ModuleVisitor::visit_module or ModuleVisitor::visit_tensor for each field, and returns optionally constructed fields