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

Example

linear = ch.nn.Linear(23, 5, bias=True)
action_mean = linear(state)