parallel_ctx_

align with pytorch: `c10/core/WrapDimMinimal.h`

r"""Copies module's states and other graph states from :attr:`state_dict` into this graph. If :attr:`strict` is ``True``, then the key

r""" zero_grad(set_to_none=False) Sets gradients of all model parameters to zero. See similar function under :class:`

Return last computed learning rate by current scheduler.

Getters

Returns a new distribution instance (or populates an existing instance provided by a derived class) with batch dimensions expanded to

Return an iterable of the ModuleDict keys.

r""" register_parameter(name, param) Adds a parameter to the module. The parameter can be accessed as an attribute u

Generate a arithmetic progression with given size and start value.

Setters