Shortcuts

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 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.

log_prob(value: Tensor) Tensor[source]

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(size=None) Tensor[source]

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

sample(size=None) Tensor[source]

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources