syntaxdot_transformers::module

Trait FallibleModule

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

    // Required method
    fn forward(&self, input: &Tensor) -> 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(&self, input: &Tensor) -> Result<Tensor, Self::Error>

Apply the module.

Implementors§