tensordict.nn.distributions.Delta

class tensordict.nn.distributions.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) – 絕對容忍度，用於判斷 tensor 是否與分布參數相符; 預設值為 1e-6

• rtol (number, optional) – 相對容忍度，用於判斷 tensor 是否與分布參數相符; 預設值為 1e-6

• batch_shape (torch.Size, optional) – 批次形狀;

• event_shape (torch.Size, optional) – 結果的形狀。