ruPRIM-host-0.1.45
- ruPRIM-host 0.1.45
- Permalink
- Docs.rs crate page
- MIT OR Apache-2.0
- 07 October 2026
- Links
- Repository
- crates.io
- Source
- Owners
- shuqi2077
- Dependencies
- aligned-vec ^0.6 normal optional
  bytemuck ^1.25.0 normal
  half ^2.7.1 normal
  libm ^0.2.15 normal
  macerator ^0.3.0 normal optional
  num-traits ^0.2.19 normal
  rayon ^1.10.0 normal optional
  ruda-core ^0.1.5 normal
- Versions
- 37.2% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

Skip to main content

float_argmax

ruprim_host0.1.45

In ruprim_host::reduce::dispatch

ruprim_host::reduce::dispatch

Function float_argmax

pub fn float_argmax(
    tensor: HostTensor,
    dim: usize,
    out_dtype: IntDType,
) -> HostTensor