syntaxdot_transformers::module

Trait FallibleModuleT

Source 
pub trait FallibleModuleT: Debug + Send {
    type Error;

    // Required method
    fn forward_t(
        &self,
        input: &Tensor,
        train: bool,
    ) -> Result<Tensor, Self::Error>;
}
Expand description

Module for which a computation can fail.

Required Associated Types§

Source

type Error

The error type.

Required Methods§

Source

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

Apply the module.

Implementors§