torch.nn.parallel._functions.Scatter Class Reference

Inheritance diagram for torch.nn.parallel._functions.Scatter:

Static Public Member Functions
def	forward (ctx, target_gpus, chunk_sizes, dim, input)
def	backward (ctx, grad_output)

Detailed Description

Definition at line 78 of file _functions.py.

---

The documentation for this class was generated from the following file:

• torch/nn/parallel/_functions.py