torch.nn.parallel.data_parallel.DataParallel Class Reference

Inheritance diagram for torch.nn.parallel.data_parallel.DataParallel:

Public Member Functions
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)

Public Attributes
	module
	device_ids
	dim
	output_device
	src_device_obj

Detailed Description

Definition at line 36 of file data_parallel.py.

---

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

• torch/nn/parallel/data_parallel.py