rai-core-0.9.0
- rai-core 0.9.0
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- cksac
- Dependencies
- candle-core ^0.3.3 normal optional
  
  candle-flash-attn ^0.3.3 normal optional
  
  colored ^2.1.0 normal
  
  dyn-clone ^1.0.16 normal
  
  half ^2.3.1 normal
  
  num-traits ^0.2.17 normal
  
  once_cell ^1.19.0 normal
  
  paste ^1.0 normal
  
  safetensors ^0.4.1 normal
  
  thiserror ^1.0.51 normal
  
  tracing ^0.1.40 normal
- Versions
- 12.33% of the crate is documented
Go to latest version
Platform
Feature flags

docs.rs

Rust

rai_core0.9.0

In rai_core::ops

Function rai_core::ops::softmax

pub fn softmax<D: Dim>(x: &Tensor, d: D) -> Tensor