tch-plus-0.18.3
- tch-plus 0.18.3
- Permalink
- Docs.rs crate page
- MIT/Apache-2.0
- Links
- Repository
- crates.io
- Source
- Owners
- jacksors
- Dependencies
- clap ^4.2.4 normal optional
  cpython ^0.7.1 normal optional
  half ^2 normal
  image ^0.24.5 normal optional
  lazy_static ^1.3.0 normal
  libc ^0.2.0 normal
  memmap2 ^0.6.1 normal optional
  ndarray ^0.15 normal
  rand ^0.8 normal
  regex ^1.6.0 normal optional
  safetensors ^0.3.0 normal
  serde_json ^1.0.96 normal optional
  thiserror ^1 normal
  torch-sys-plus ^0.18.2 normal
  zip ^0.6 normal
  anyhow ^1.0.60 dev
- Versions
- 37.34% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

set_num_threads

tch_plus0.18.3

In tch_plus::utils

tch_plus::utils

Function set_num_threads

pub fn set_num_threads(n_threads: i32)

Expand description

Set the number of threads used by torch in parallel regions.