Struct Sequential 

pub struct Sequential { /* private fields */ }

Builder for constructing models layer by layer.

Implementations

impl Sequential

pub fn new(name: &str) -> Self

pub fn dense(self, in_features: usize, out_features: usize) -> Self

pub fn conv2d( self, in_channels: usize, out_channels: usize, kernel_size: usize, ) -> Self

pub fn max_pool(self, kernel_size: usize, stride: usize) -> Self

pub fn batch_norm(self, num_features: usize) -> Self

pub fn dropout(self, p: f32) -> Self

pub fn flatten(self) -> Self

pub fn relu(self) -> Self

pub fn sigmoid(self) -> Self

pub fn tanh_act(self) -> Self

pub fn softmax(self) -> Self

pub fn softmax_axis(self, axis: usize) -> Self

pub fn gelu(self) -> Self

pub fn residual(self, layers: Vec<Layer>) -> Self

pub fn attention(self, heads: usize, d_model: usize) -> Self

pub fn layer(self, layer: Layer) -> Self

pub fn build(self) -> Model