RoboticsLinear

RoboticsLinear(*args, **kwargs)

[Source]

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)