rai-core-0.7.0
- rai-core 0.7.0
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- cksac
- Dependencies
- candle-core ^0.3.2 normal
  
  colored ^2.1.0 normal
  
  dyn-clone ^1.0.16 normal
  
  half ^2.3.1 normal
  
  once_cell ^1.19.0 normal
  
  paste ^1.0 normal
  
  safetensors ^0.3.1 normal
  
  thiserror ^1.0.51 normal
  
  tracing ^0.1.40 normal
- Versions
- 0% of the crate is documented
Go to latest version
Platform
Feature flags

docs.rs

Rust

rai_core0.7.0

In rai_core::ops

Function rai_core::ops::argmax

pub fn argmax<T: ArgReduceArgs>(x: &Tensor, args: T) -> Tensor