Struct dfdx::nn::Repeated

pub struct Repeated<T, const N: usize> {
    pub modules: Vec<T>,
}

Repeats T N times. This requires that T’s input is the same as it’s output.

Generics

• T the Module to repeat
• N the number of times to repeat T.

Examples

type Model = Repeated<(Linear<10, 10>, ReLU), 5>;
let model: Model = Default::default();
let out: Tensor1D<10> = model.forward(Tensor1D::zeros());

Fields

modules: Vec<T>

Trait Implementations

impl<T: CanUpdateWithGradients, const N: usize> CanUpdateWithGradients for Repeated<T, N>

fn update<G: GradientProvider>(
    &mut self,
    grads: &mut G,
    unused: &mut UnusedTensors
)

Updates self given the GradientProvider. When any parameters that are NOT present in G, then this function should add the tensor’s UniqueId to UnusedTensors.

impl<T: Clone, const N: usize> Clone for Repeated<T, N>

fn clone(&self) -> Repeated<T, N>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug, const N: usize> Debug for Repeated<T, N>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<T: Default, const N: usize> Default for Repeated<T, N>

fn default() -> Self

Returns the “default value” for a type.

impl<T, const N: usize> Index<usize> for Repeated<T, N>

type Output = T

The returned type after indexing.

fn index(&self, index: usize) -> &Self::Output

Performs the indexing (container[index]) operation.

impl<T: LoadFromNpz, const N: usize> LoadFromNpz for Repeated<T, N>

fn read<R: Read + Seek>(
    &mut self,
    p: &str,
    r: &mut ZipArchive<R>
) -> Result<(), NpzError>

Reads this object from a ZipArchive. r with a base filename of filename_prefix.

fn load<P: AsRef<Path>>(&mut self, path: P) -> Result<(), NpzError>

Loads data from a .npz zip archive at the specified path.

impl<Input, T: Module<Input, Output = Input>, const N: usize> Module<Input> for Repeated<T, N>

type Output = <T as Module<Input>>::Output

The type that this unit produces given Input.

fn forward(&self, x: Input) -> Self::Output

Forward Input through the module and produce Module::Output.

impl<Input, T: ModuleMut<Input, Output = Input>, const N: usize> ModuleMut<Input> for Repeated<T, N>

type Output = <T as ModuleMut<Input>>::Output

The type that this unit produces given Input.

fn forward_mut(&mut self, x: Input) -> Self::Output

Forward Input through the module and produce ModuleMut::Output.

impl<T: ResetParams, const N: usize> ResetParams for Repeated<T, N>

fn reset_params<R: Rng>(&mut self, rng: &mut R)

Mutate the unit’s parameters using rand::Rng. Each implementor of this trait decides how the parameters are initialized. In fact, some impls may not even use the rng.

impl<T: SaveToNpz, const N: usize> SaveToNpz for Repeated<T, N>

fn write<W: Write + Seek>(&self, p: &str, w: &mut ZipWriter<W>) -> ZipResult<()>

Write this object into ZipWriter w with a base filename of filename_prefix.

fn save<P: AsRef<Path>>(&self, path: P) -> ZipResult<()>

Save this object into the .npz file determined located at path.