1
2
3
4
5
6
7
8
9
10
11
12
13
//! 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.

mod collection;
mod visitor;
mod visitor_impls;

pub use collection::{ModuleVisitor, ScalarOptions, TensorCollection, TensorOptions};
pub use visitor::{
    ModuleField, ModuleFields, RecursiveWalker, ScalarField, TensorField, TensorViewer,
    TensorVisitor, ViewTensorMut, ViewTensorName, ViewTensorRef,
};