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) – 結果的形狀。
- expand(batch_shape: Size, _instance=None)[source]¶
返回一個新的分佈例項(或填充派生類提供的現有例項),其批次維度已擴充套件到 batch_shape。此方法對分佈的引數呼叫
expand。因此,這不會為擴充套件後的分佈例項分配新記憶體。此外,當首次建立例項時,這不會重複 __init__.py 中的任何引數檢查或引數廣播。- 引數:
batch_shape (torch.Size) – 期望的擴充套件大小。
_instance – 需要覆蓋 .expand 的子類提供的新例項。
- 返回值:
批次維度擴充套件到 batch_size 的新分佈例項。