Enum StorageWrap

pub enum StorageWrap {
    NdArray(NdArray),
    QuantizedArray(QuantizedArray),
    MmapArray(MmapArray),
}

Storage types wrapper.

This crate makes it possible to create fine-grained embedding types, such as Embeddings<SimpleVocab, NdArray> or Embeddings<SubwordVocab, QuantizedArray>. However, in some cases it is more pleasant to have a single type that covers all vocabulary and storage types. VocabWrap and StorageWrap wrap all the vocabularies and storage types known to this crate such that the type Embeddings<VocabWrap, StorageWrap> covers all variations.

Variants

NdArray(NdArray)

QuantizedArray(QuantizedArray)

MmapArray(MmapArray)

Trait Implementations

impl From<MmapArray> for StorageWrap

fn from(s: MmapArray) -> Self

Converts to this type from the input type.

impl From<NdArray> for StorageWrap

fn from(s: NdArray) -> Self

Converts to this type from the input type.

impl From<QuantizedArray> for StorageWrap

fn from(s: QuantizedArray) -> Self

Converts to this type from the input type.

impl Storage for StorageWrap

fn embedding(&self, idx: usize) -> CowArray1<'_, f32>

fn shape(&self) -> (usize, usize)