Struct tch::nn::SequentialT

pub struct SequentialT { /* private fields */ }

A sequential layer combining new layers with support for a training mode.

Implementations

impl SequentialT

pub fn len(&self) -> i64

The number of sub-layers embedded in this layer.

pub fn is_empty(&self) -> bool

Returns true if this layer does not have any sub-layer.

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) -> Selfwhere F: 'static + Fn(&Tensor) -> Tensor + Send,

Appends a closure after all the current layers.

pub fn add_fn_t<F>(self, f: F) -> Selfwhere F: 'static + Fn(&Tensor, bool) -> Tensor + Send,

Appends a closure after all the current layers.

pub fn forward_all_t( &self, xs: &Tensor, train: bool, n: Option<usize> ) -> Vec<Tensor>

Applies the forward pass and returns the output for each layer.

Trait Implementations

impl Debug for SequentialT

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

Formats the value using the given formatter.

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