Index

C

clone() (torch_blue.vi.VIReturn method)

convert_to_vimodule() (in module torch_blue.vi)

F

forward() (torch_blue.vi.AnalyticalKullbackLeiblerLoss method) (torch_blue.vi.KullbackLeiblerLoss method) (torch_blue.vi.NonBayesianDivergence static method) (torch_blue.vi.NormalNormalDivergence static method) (torch_blue.vi.UniformNormalDivergence static method) (torch_blue.vi.VIConv1d method) (torch_blue.vi.VIConv2d method) (torch_blue.vi.VIConv3d method) (torch_blue.vi.VILinear method) (torch_blue.vi.VIMultiheadAttention method) (torch_blue.vi.VITransformer method) (torch_blue.vi.VITransformerDecoder method) (torch_blue.vi.VITransformerDecoderLayer method) (torch_blue.vi.VITransformerEncoder method) (torch_blue.vi.VITransformerEncoderLayer method) (VIResidualConnection method) (VISequential method)

from_tensor() (torch_blue.vi.VIReturn static method)

G

gather_log_probs() (torch_blue.vi.VIModule method)

get_log_probs() (torch_blue.vi.VIModule method) get_variational_parameters() (torch_blue.vi.VIModule method)

K

kaiming_rescale() (torch_blue.vi.distributions.Distribution method)

L

log_prob_from_parameters() (torch_blue.vi.distributions.Categorical static method) (torch_blue.vi.distributions.MeanFieldNormal method) (torch_blue.vi.distributions.NonBayesian method)

log_prob_from_samples() (torch_blue.vi.distributions.Distribution method)

M

match_parameters() (torch_blue.vi.distributions.Distribution method) module torch_blue torch_blue.vi torch_blue.vi.distributions torch_blue.vi.utils

N

new_empty() (torch_blue.vi.VIReturn method)

P

predictive_parameters_from_samples() (torch_blue.vi.distributions.Categorical method) (torch_blue.vi.distributions.MeanFieldNormal static method) (torch_blue.vi.distributions.NonBayesian static method) primary_parameter (torch_blue.vi.distributions.Distribution property)

prior_log_prob() (torch_blue.vi.distributions.BasicQuietPrior method) (torch_blue.vi.distributions.MeanFieldNormal method) (torch_blue.vi.distributions.NonBayesian static method) prior_matching() (torch_blue.vi.AnalyticalKullbackLeiblerLoss method)

R

random_variables (torch_blue.vi.VIModule property) reset_log_probs() (torch_blue.vi.VIModule method) reset_parameters_to_prior() (torch_blue.vi.distributions.BasicQuietPrior method) (torch_blue.vi.distributions.Distribution method) (torch_blue.vi.distributions.MeanFieldNormal method)

reset_variational_parameters() (torch_blue.vi.distributions.Distribution method) (torch_blue.vi.VIModule method) return_log_probs (torch_blue.vi.VIModule property) (torch_blue.vi.VITransformerDecoder attribute) (torch_blue.vi.VITransformerDecoderLayer attribute) (torch_blue.vi.VITransformerEncoderLayer attribute)

S

sample() (torch_blue.vi.distributions.MeanFieldNormal method) (torch_blue.vi.distributions.MeanFieldStudentT method) (torch_blue.vi.distributions.NonBayesian method)

sample_variable() (torch_blue.vi.VIModule method) sampled_forward() (torch_blue.vi.VIModule method) std (torch_blue.vi.distributions.MeanFieldNormal property)

T

to() (torch_blue.vi.VIReturn method) torch_blue module torch_blue.vi module torch_blue.vi.AnalyticalKullbackLeiblerLoss (built-in class) torch_blue.vi.distributions module torch_blue.vi.distributions.BasicQuietPrior (built-in class) torch_blue.vi.distributions.Categorical (built-in class) torch_blue.vi.distributions.Distribution (built-in class) torch_blue.vi.distributions.MeanFieldNormal (built-in class) torch_blue.vi.distributions.MeanFieldStudentT (built-in class) torch_blue.vi.distributions.NonBayesian (built-in class) torch_blue.vi.distributions.UniformPrior (built-in class) torch_blue.vi.KullbackLeiblerLoss (built-in class) torch_blue.vi.KullbackLeiblerModule (built-in class) torch_blue.vi.NonBayesianDivergence (built-in class) torch_blue.vi.NormalNormalDivergence (built-in class) torch_blue.vi.UniformNormalDivergence (built-in class) torch_blue.vi.utils module

torch_blue.vi.utils.NoVariablesError torch_blue.vi.utils.PostInitCallMeta (built-in class) torch_blue.vi.utils.UnsupportedDistributionError torch_blue.vi.VIConv1d (built-in class) torch_blue.vi.VIConv2d (built-in class) torch_blue.vi.VIConv3d (built-in class) torch_blue.vi.VIkwargs (built-in class) torch_blue.vi.VILinear (built-in class) torch_blue.vi.VIModule (built-in class) torch_blue.vi.VIMultiheadAttention (built-in class) torch_blue.vi.VIResidualConnection (built-in class) torch_blue.vi.VIReturn (built-in class) torch_blue.vi.VISequential (built-in class) torch_blue.vi.VITransformer (built-in class) torch_blue.vi.VITransformerDecoder (built-in class) torch_blue.vi.VITransformerDecoderLayer (built-in class) torch_blue.vi.VITransformerEncoder (built-in class) torch_blue.vi.VITransformerEncoderLayer (built-in class) track() (torch_blue.vi.AnalyticalKullbackLeiblerLoss method) (torch_blue.vi.KullbackLeiblerLoss method)

U

use_norm_constants() (in module torch_blue.vi.utils)

V

variational_log_prob() (torch_blue.vi.distributions.MeanFieldNormal method) (torch_blue.vi.distributions.MeanFieldStudentT method) (torch_blue.vi.distributions.NonBayesian method)

variational_parameter_name() (torch_blue.vi.VIModule static method) VIResidualConnection (built-in class) VISequential (built-in class)