Trait AsModule

pub trait AsModule {
    type Module: Module;

    // Required methods
    fn as_module(&self) -> &Self::Module;
    fn as_module_mut(&mut self) -> &mut Self::Module;

    // Provided method
    fn batch_map<F: Fn(Tensor) -> Tensor>(self, f: F) -> BatchMap<Self, F>
       where Self: Sized { ... }
}

A trait for module wrappers that allow borrowing the contained Module.

Required Associated Types

type Module: Module

Required Methods

fn as_module(&self) -> &Self::Module

fn as_module_mut(&mut self) -> &mut Self::Module

Provided Methods

fn batch_map<F: Fn(Tensor) -> Tensor>(self, f: F) -> BatchMap<Self, F>

where Self: Sized,

Apply a batch-structure-preserving function to the module output.

Implementors

impl<M: Module, O> AsModule for Ppo<M, O>

type Module = M

impl<M: Module, O> AsModule for Reinforce<M, O>

type Module = M

impl<M: Module, O> AsModule for Trpo<M, O>

type Module = M

impl<T: AsModule> AsModule for WithCpuCopy<T>

type Module = <T as AsModule>::Module

impl<T: AsModule, F> AsModule for BatchMap<T, F>

type Module = <T as AsModule>::Module

impl<T: Module> AsModule for T

type Module = T