nami.distributions.base

Functions

expand_distribution(dist, batch_shape)
has_rsample(dist)

nami.distributions.base.expand_distribution(dist, batch_shape)

Return type:

Distribution

Parameters:

• dist (Distribution)

• batch_shape (tuple[int, ...])

nami.distributions.base.has_rsample(dist)

Return type:

bool

Parameters:

dist (Distribution)