axonml-distributed-0.2.2
- axonml-distributed 0.2.2
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- crates.io
- Source
- Owners
- AutomataControls
- Dependencies
- axonml-autograd ^0.2.1 normal
  axonml-core ^0.2.1 normal
  axonml-nn ^0.2.1 normal
  axonml-tensor ^0.2.1 normal
- Versions
- 100% of the crate is documented
Go to latest version
Platform
Feature flags

docs.rs

Rust

sync_gradient

axonml_distributed0.2.2

In axonml_distributed::comm

axonml_distributed::comm

Function sync_gradient

pub fn sync_gradient(gradient: &mut Tensor<f32>, pg: &ProcessGroup)

Expand description

Synchronizes a single gradient tensor.