Delta

class torchrl.modules.Delta(param: Tensor, atol: float = 1e-06, rtol: float = 1e-06, batch_shape: Optional[Union[Size, Sequence[int]]] = None, event_shape: Optional[Union[Size, Sequence[int]]] = None)

Delta 分布。

参数:

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

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

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

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

• event_shape (torch.Size, optional) – 结果的形状。

log_prob(value: Tensor) → Tensor

返回在 value 处评估的概率密度/质量函数的对数。

参数:

value (Tensor) –

property mean: Tensor

返回分布的均值。

property mode: Tensor

返回分布的众数。

rsample(size=None) → Tensor

生成 sample_shape 形状的重参数化样本，或者如果分布参数是批处理的，则生成 sample_shape 形状的重参数化样本批次。

sample(size=None) → Tensor

生成 sample_shape 形状的样本，或者如果分布参数是批处理的，则生成 sample_shape 形状的样本批次。