Struct candle_nn::sequential::Sequential

pub struct Sequential { /* private fields */ }

A sequential layer combining multiple other layers.

Implementations

impl Sequential

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 Sequential

pub fn add<M: Module + '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 + Fn(&Tensor) -> Result<Tensor> + Send + Sync,

Appends a closure after all the current layers.

pub fn forward_all(&self, xs: &Tensor) -> Result<Vec<Tensor>>

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

Trait Implementations

impl Module for Sequential

fn forward(&self, xs: &Tensor) -> Result<Tensor>