diffusion_rs_common-0.1.0
- diffusion_rs_common 0.1.0
- Permalink
- Docs.rs crate page
- MIT
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- EricLBuehler
- Dependencies
- accelerate-src ^0.3.2 normal optional
  anyhow ^1.0.94 normal
  byteorder ^1.5.0 normal
  cudarc ^0.12.1 normal optional
  dirs ^5.0.1 normal
  float8 ^0.1.0 normal
  gemm ^0.17.0 normal
  half ^2.4.1 normal
  hf-hub ^0.3.2 normal
  indicatif ^0.17.9 normal
  intel-mkl-src ^0.8.1 normal optional
  libc ^0.2.147 normal optional
  memmap2 ^0.9.5 normal
  metal ^0.27.0 normal optional
  num-traits ^0.2.15 normal
  num_cpus ^1.15.0 normal
  rand ^0.8.5 normal
  rand_distr ^0.4.3 normal
  rayon ^1.1.0 normal
  safetensors ^0.4.1 normal
  serde ^1.0.215 normal
  serde_json ^1.0.133 normal
  thiserror ^2.0.4 normal
  tokenizers ^0.21.0 normal
  tqdm ^0.7.0 normal
  ug ^0.0.2 normal
  ug-cuda ^0.0.2 normal optional
  ug-metal ^0.0.2 normal optional
  yoke ^0.7.2 normal
  zip ^2.2.1 normal
  bindgen_cuda ^0.1.5 build optional
- Versions
- 17.76% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

log_softmax

diffusion_rs_common0.1.0

In diffusion_rs_common::nn::ops

diffusion_rs_common::nn::ops

Function log_softmax

pub fn log_softmax<D: Dim>(xs: &Tensor, d: D) -> Result<Tensor>