Struct tch::nn::SequentialT

pub struct SequentialT { /* fields omitted */ }

Methods

impl SequentialT

pub fn new() -> SequentialT

impl SequentialT

pub fn add<M: ModuleT + 'static>(self, layer: M) -> Self

Appends a layer after all the current layers.

pub fn add_fn<F>(self, f: F) -> Self where
    F: 'static,
    F: Fn(&Tensor) -> Tensor, 

Appends a closure after all the current layers.

pub fn add_fn_t<F>(self, f: F) -> Self where
    F: 'static,
    F: Fn(&Tensor, bool) -> Tensor, 

Appends a closure after all the current layers.

Trait Implementations

impl ModuleT for SequentialT

fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor

fn batch_accuracy_for_logits(
    &self,
    xs: &Tensor,
    ys: &Tensor,
    d: Device,
    batch_size: i64
) -> f64