tensordict.nn.distributions.Delta

class tensordict.nn.distributions.Delta(param: torch.Tensor, atol: float = 1e-06, rtol: float = 1e-06, batch_shape: torch.Size | Sequence[int] | None = None, event_shape: torch.Size | Sequence[int] | None = None)

Delta distribution.

Parameters:

• param (torch.Tensor) – parameter of the delta distribution;

• atol (number, optional) – absolute tolerance to consider that a tensor matches the distribution parameter; Default is 1e-6

• rtol (number, optional) – relative tolerance to consider that a tensor matches the distribution parameter; Default is 1e-6

• batch_shape (torch.Size, optional) – batch shape;

• event_shape (torch.Size, optional) – shape of the outcome.