TanhDelta

class torchrl.modules.TanhDelta(param: torch.Tensor, low: torch.Tensor | float = - 1.0, high: torch.Tensor | float = 1.0, event_dims: int = 1, atol: float = 1e-06, rtol: float = 1e-06, safe: bool = True)

实现了 Tanh 变换的 Delta 分布。

参数:

• param (torch.Tensor) – delta 分布的参数；

• low (torch.Tensor 或 数字, 可选) – 分布的最小值。默认为 -1.0；

• high (torch.Tensor 或 数字, 可选) – 分布的最大值。默认为 1.0；

• event_dims (int, 可选) – 描述动作的维度数。默认为 1；

• atol (数字, 可选) – 用于判断张量是否匹配分布参数的绝对容差；默认为 1e-6

• rtol (数字, 可选) – 用于判断张量是否匹配分布参数的相对容差；默认为 1e-6

• batch_shape (torch.Size, 可选) – 批次形状；

• event_shape (torch.Size, 可选) – 输出形状；

property mean: Tensor

返回分布的均值。

property mode: Tensor

返回分布的众数。