Skip to main content

AutoregressiveModel

deep_delta_learning::generation

Trait AutoregressiveModel 

Source 
pub trait AutoregressiveModel<B: Backend> {
    // Required methods
    fn forward_logits(
        &self,
        input_ids: Tensor<B, 2, Int>,
        mask: Option<&Tensor<B, 3>>,
    ) -> Tensor<B, 3>;
    fn max_seq_len(&self) -> usize;
}

Required Methods§

Source

fn forward_logits( &self, input_ids: Tensor<B, 2, Int>, mask: Option<&Tensor<B, 3>>, ) -> Tensor<B, 3>

Source

fn max_seq_len(&self) -> usize

Implementors§