reference/generated/tensordict.nn.distributions.Delta

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

Delta

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

Bases: Distribution

Delta distribution.

A distribution with all its mass on param. sample(), rsample(), mean, mode and deterministic_sample return param. log_prob() returns inf for a value equal to param (within atol and rtol) and -inf otherwise. It is the default distribution_class of ProbabilisticTensorDictModule.

Parameters:

• param (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 (Size, optional) – batch shape;

• event_shape (Size, optional) – shape of the outcome. If neither batch_shape nor event_shape is given, the last dimension of param is the event shape and the others are the batch shape.

Examples

>>> import torch
>>> from tensordict.nn.distributions import Delta
>>> dist = Delta(torch.zeros(4, 2))
>>> dist.batch_shape, dist.event_shape
(torch.Size([4]), torch.Size([2]))
>>> dist = Delta(torch.tensor([1.0, 2.0]))
>>> dist.sample()
tensor([1., 2.])
>>> dist.log_prob(torch.tensor([1.0, 2.0])), dist.log_prob(torch.tensor([1.0, 3.0]))
(tensor(inf), tensor(-inf))

log_prob(value: Tensor) → Tensor

Returns the log of the probability density/mass function evaluated at value.

Parameters:

value (Tensor) –

property mean: Tensor

Returns the mean of the distribution.

property mode: Tensor

Returns the mode of the distribution.

rsample(sample_shape: Sequence[int] | Size | None = None) → Tensor

Generates a sample_shape shaped reparameterized sample or sample_shape shaped batch of reparameterized samples if the distribution parameters are batched.

sample(sample_shape: Sequence[int] | Size | None = None) → Tensor

Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched.