ghostflow-cuda-1.0.0
- ghostflow-cuda 1.0.0
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- choksi2212
- Dependencies
- ghostflow-core ^1.0.0 normal
  parking_lot ^0.12 normal
  cc ^1.0 build
- Versions
- 65.1% of the crate is documented
Platform
Feature flags

docs.rs

Rust

softmax

ghostflow_cuda1.0.0

In ghostflow_cuda::ops::elementwise

ghostflow_cuda::ops::elementwise

Function softmax

pub fn softmax(
    x: &CudaTensor,
    dim: i32,
    stream: &CudaStream,
) -> CudaResult<CudaTensor>

Expand description

Softmax