candle-nn-0.5.0
- candle-nn 0.5.0
- Permalink
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Repository
- Crates.io
- Source
- Owners
- LaurentMazare
- Narsil
- Dependencies
- accelerate-src ^0.3.2 normal optional
  candle-core ^0.5.0 normal
  candle-metal-kernels ^0.5.0 normal optional
  half ^2.3.1 normal
  intel-mkl-src ^0.8.1 normal optional
  metal ^0.27.0 normal optional
  num-traits ^0.2.15 normal
  rayon ^1.7.0 normal
  safetensors ^0.4.1 normal
  serde ^1.0.171 normal
  thiserror ^1 normal
  anyhow ^1 dev
  clap ^4.2.4 dev
  criterion ^0.5.1 dev
  rand ^0.8.5 dev
- Versions
- 33.64% of the crate is documented
Platform
Feature flags

Rust

candle_nn0.5.0

In candle_nn::embedding

Function candle_nn::embedding::embedding

pub fn embedding(
    in_size: usize,
    out_size: usize,
    vb: VarBuilder<'_>
) -> Result<Embedding>