快捷方式

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)[source]

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[source]

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

参数::

value (Tensor) –

property mean: Tensor

返回分布的均值。

property mode: Tensor

返回分布的众数。

rsample(size=None) Tensor[source]

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

sample(size=None) Tensor[source]

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

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源