Module dfdx::nn::tensor_collection

Traits to define a TensorCollection and how to iterate them using ModuleVisitor. Use RecursiveWalker to do the iteration and TensorVisitor to define functions to iterate through and/or construct Modules.

Structs

• ModuleField
  A ModuleFields that represents a field that contains one or more Tensors.
• RecursiveWalker
  A standard ModuleVisitor that executes F on every Tensor encountered. F must implement TensorVisitor
• ScalarField
  A ModuleFields that represents a field that contains a scalar value that should be serialized.
• ScalarOptions
  Options to change behavior of ModuleVisitor
• TensorField
  A ModuleFields that represents a field that contains a single Tensor.
• TensorOptions
  Options to change behavior of ModuleVisitor

Enums

• ViewTensorMut
  A TensorViewer that represents a &mut Tensor
• ViewTensorName
  A TensorViewer that represents a Tensor’s name as a String
• ViewTensorRef
  A TensorViewer that represents a &Tensor

Traits

• ModuleFields
  A list of a Module’s fields. Used in ModuleVisitor::visit_fields.
• ModuleVisitor
  An object that can visit TensorCollections and Tensors recursively.
• TensorCollection
  A collection of named tensors. Implementing this trait will enable anything that operates on tensors, including resetting, counting number of params, updating gradients, building model, and converting models between devices or dtypes.
• TensorViewer
  Something that can view Tensors in different ways. For example ViewTensorRef can view &Tensor, and ViewTensorMut can view `&mut Tensor.
• TensorVisitor
  Something that can visit Tensors. Used in conjunction with RecursiveWalker.