Index

A

arg_constraints (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli attribute) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli attribute) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK attribute) (relaxit.distributions.HardConcrete.HardConcrete attribute) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian attribute) (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax attribute)

B

batch_shape (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli property) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli property) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK property) (relaxit.distributions.HardConcrete.HardConcrete property) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian property)

C

CorrelatedRelaxedBernoulli (class in relaxit.distributions.CorrelatedRelaxedBernoulli)

D

dirichlet_approximation_fn() (in module relaxit.distributions.approx)

E

event_shape (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli property) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli property) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK property) (relaxit.distributions.HardConcrete.HardConcrete property) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian property)

expand() (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax method)

G

GaussianRelaxedBernoulli (class in relaxit.distributions.GaussianRelaxedBernoulli)

GumbelSoftmaxTopK (class in relaxit.distributions.GumbelSoftmaxTopK)

H

HardConcrete (class in relaxit.distributions.HardConcrete) has_rsample (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli attribute) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli attribute) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK attribute) (relaxit.distributions.HardConcrete.HardConcrete attribute) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian attribute) (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax attribute) (relaxit.distributions.StochasticTimesSmooth.StochasticTimesSmooth attribute) (relaxit.distributions.StraightThroughBernoulli.StraightThroughBernoulli attribute)

I

InvertibleGaussian (class in relaxit.distributions.InvertibleGaussian)

L

loc (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax property) log_prob() (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli method) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli method) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK method) (relaxit.distributions.HardConcrete.HardConcrete method) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian method)

LogisticNormalSoftmax (class in relaxit.distributions.LogisticNormalSoftmax) lognorm_approximation_fn() (in module relaxit.distributions.approx)

M

module relaxit relaxit.distributions.approx relaxit.distributions.kl

R

relaxit module relaxit.distributions.approx module relaxit.distributions.kl module

rsample() (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli method) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli method) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK method) (relaxit.distributions.HardConcrete.HardConcrete method) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian method) (relaxit.distributions.StochasticTimesSmooth.StochasticTimesSmooth method) (relaxit.distributions.StraightThroughBernoulli.StraightThroughBernoulli method)

S

sample() (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli method) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli method) (relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK method) (relaxit.distributions.HardConcrete.HardConcrete method) scale (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax property) softmax_plus_plus() (relaxit.distributions.InvertibleGaussian.InvertibleGaussian method)

StochasticTimesSmooth (class in relaxit.distributions.StochasticTimesSmooth) StraightThroughBernoulli (class in relaxit.distributions.StraightThroughBernoulli) support (relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli attribute) (relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli attribute) (relaxit.distributions.HardConcrete.HardConcrete attribute) (relaxit.distributions.InvertibleGaussian.InvertibleGaussian attribute) (relaxit.distributions.LogisticNormalSoftmax.LogisticNormalSoftmax attribute)