volta-0.3.0
- volta 0.3.0
- Permalink
- Docs.rs crate page
- MIT
- Links
- Repository
- crates.io
- Source
- Owners
- rlarson20
- Dependencies
- approx ^0.5.1 normal
  bincode ^2.0.1 normal
  bytemuck ^1.24.0 normal
  cblas-sys ^0.3.0 normal
  half ^2.7.1 normal
  matrixmultiply ^0.3.10 normal
  pollster ^0.4.0 normal optional
  rand ^0.9.2 normal
  rand_distr ^0.5.1 normal
  safetensors ^0.7.0 normal
  serde_json ^1.0.149 normal
  wgpu ^27.0.1 normal optional
  criterion ^0.5 dev
  blas-src ^0.14 normal optional
- Versions
- 53.97% of the crate is documented
Platform
Feature flags

docs.rs

Rust

softmax

volta0.3.0

In volta::tensor

Function softmax

pub fn softmax(t: &Tensor, dim: usize) -> Tensor