torch.nn.parallel._functions.Broadcast Class Reference

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

Static Public Member Functions
def	forward (ctx, target_gpus, inputs)
def	backward (ctx, grad_outputs)

Detailed Description

Definition at line 9 of file _functions.py.

---

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

• torch/nn/parallel/_functions.py