torsh-ffi-0.1.0-alpha.2
- torsh-ffi 0.1.0-alpha.2
- Permalink
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- cool-japan
- Dependencies
- anyhow ^1.0 normal
  bincode ^2.0 normal
  chrono ^0.4 normal
  fastrand ^2.3 normal
  num_cpus ^1.17 normal
  numpy ^0.27 normal optional
  once_cell ^1.21 normal
  parking_lot ^0.12 normal
  pyo3 ^0.27 normal optional
  scirs2-core ^0.1.0-rc.4 normal
  serde ^1.0 normal
  serde_json ^1.0 normal
  thiserror ^2.0 normal
  torsh-core ^0.1.0-alpha.2 normal
  torsh-data ^0.1.0-alpha.2 normal
  torsh-nn ^0.1.0-alpha.2 normal
  torsh-optim ^0.1.0-alpha.2 normal
  torsh-tensor ^0.1.0-alpha.2 normal
  tracing ^0.1 normal
  uuid ^1.19 normal
  walkdir ^2.5 normal
  tempfile ^3.23 dev
  pyo3-build-config ^0.27 build
- Versions
- 70.24% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

softmax

torsh_ffi0.1.0-alpha.2

In torsh_ffi::python

torsh_ffi::python

Function softmax

pub fn softmax(input: &PyTensor, _dim: i32) -> PyResult<PyTensor>

Expand description

Softmax function