oxidized-transformers 0.1.1

Transformers library (not functional yet)
Documentation 
1
2
3
4
5
6
7
8
9
10

use candle_core::Tensor;

/// Trait for querying layer outputs of a model.
pub trait LayerOutputs {
    /// Outputs of all layers.
    fn layer_outputs(&self) -> &[Tensor];

    /// Output of the embedding layer.
    fn embedding_layer_output(&self) -> Option<&Tensor>;
}