drug-0.0.2
- drug 0.0.2
- Docs.rs crate page
- MIT
- Links
- Repository
- Crates.io
- Source
- Owners
- CasperN
- Dependencies
- debug_stub_derive ^0.3.0 normal
  erased-serde ^0.3 normal
  ndarray ^0.11.2 normal
  rand ^0.5.4 normal
  serde ^1.0 normal
  serde_derive ^1.0 normal
  byteorder ^1.2.4 dev
  itertools ^0.7.8 dev
  ron ^0.4.0 dev
- Versions
- 51.28% of the crate is documented
Platform
Feature flags

Rust

In drug

Function drug::softmax

pub fn softmax(logits: &ArrayD<f32>) -> Array2<f32>

Expand description

Take the softmax of an array of shape batch_size * num_classes