def __init__(self, representation_dim: int, action_dim: int, action_bound: float, hidden_dimensions: List[int], nonlinearity: str, layernorm: bool, log_param_min: float, log_param_max: float):
    assert action_bound, 'Beta policy needs action bounds specified.'
    super().__init__(representation_dim=representation_dim, action_dim=action_dim, action_bound=action_bound, hidden_dimensions=hidden_dimensions, nonlinearity=nonlinearity, layernorm=layernorm, log_param_min=log_param_min, log_param_max=log_param_max)
    self.dist_head = nn.Linear(hidden_dimensions[(- 1)], (2 * self.action_dim))
def __init__(self, representation_dim: int, action_dim: int, action_bound: float, hidden_dimensions: List[int], nonlinearity: str, layernorm: bool, log_param_min: float, log_param_max: float):
    assert action_bound, 'Beta policy needs action bounds specified.'
    super().__init__(representation_dim=representation_dim, action_dim=action_dim, action_bound=action_bound, hidden_dimensions=hidden_dimensions, nonlinearity=nonlinearity, layernorm=layernorm, log_param_min=log_param_min, log_param_max=log_param_max)
    self.dist_head = nn.Linear(hidden_dimensions[(- 1)], (2 * self.action_dim))