rai-core-0.11.0
- rai-core 0.11.0
- Permalink
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- cksac
- Dependencies
- candle-core ^0.4.1 normal optional
  
  candle-flash-attn ^0.4.1 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
- 4.36% of the crate is documented
Platform
Feature flags

docs.rs

Rust

rai_core0.11.0

In rai_core::ops

Function softmax

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