RoboticsLinear
RoboticsLinear(*args, **kwargs)
Description
Akin to nn.Linear
, but with proper initialization for robotic control.
Credit
Adapted from Ilya Kostrikov's implementation.
Arguments
- gain (float, optional) - Gain factor passed to
robotics_init_
initialization. - This class extends
nn.Linear
and supports all of its arguments.
Example
linear = ch.nn.Linear(23, 5, bias=True)
action_mean = linear(state)