Caffe2 - Python API
A deep learning, cross platform ML framework
Static Public Member Functions | List of all members
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

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