|
def | __init__ (self, module, device_ids=None, output_device=None, dim=0) |
|
def | forward (self, inputs, kwargs) |
|
def | replicate (self, module, device_ids) |
|
def | scatter (self, inputs, kwargs, device_ids) |
|
def | parallel_apply (self, replicas, inputs, kwargs) |
|
def | gather (self, outputs, output_device) |
|
|
| module |
|
| device_ids |
|
| dim |
|
| output_device |
|
| src_device_obj |
|
Definition at line 36 of file data_parallel.py.
The documentation for this class was generated from the following file: