Struct Dropout

pub struct Dropout { /* private fields */ }

Dropout layer.

This layer zeros out random elements of a tensor with probability p. Dropout is a form of regularization and prevents co-adaptation of neurons.

Implementations

impl Dropout

pub fn new(p: f64) -> Self

Drop out elements with probability p.

Trait Implementations

impl Debug for Dropout

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

Formats the value using the given formatter.

impl FallibleModuleT for Dropout

type Error = TransformerError

The error type.

fn forward_t(&self, input: &Tensor, train: bool) -> Result<Tensor, Self::Error>

Apply the module.