Trait syntaxdot_transformers::module::FallibleModuleT

pub trait FallibleModuleT: Debug + Send {
    type Error;

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

Module for which a computation can fail.

Required Associated Types

type Error

The error type.

Required Methods

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

Apply the module.

Implementors

impl FallibleModuleT for Dropout

type Error = TransformerError

impl FallibleModuleT for VariationalDropout

type Error = TransformerError

impl FallibleModuleT for AlbertEmbeddings

type Error = TransformerError

impl FallibleModuleT for BertEmbeddings

type Error = TransformerError

impl FallibleModuleT for RobertaEmbeddings

type Error = TransformerError

impl FallibleModuleT for SinusoidalEmbeddings

type Error = TransformerError

impl<M> FallibleModuleT for Mwhere
    M: FallibleModule,

type Error = <M as FallibleModule>::Error