Pixyz
latest
Package Reference
pixyz.distributions (Distribution API)
pixyz.losses (Loss API)
pixyz.models (Model API)
pixyz.flows (Flow layers)
pixyz.utils
Pixyz
Docs
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
P
|
R
|
S
|
T
|
U
|
V
|
W
_
__init__() (pixyz.distributions.CustomProb method)
(pixyz.distributions.ElementWiseProductOfNormal method)
(pixyz.distributions.MarginalizeVarDistribution method)
(pixyz.distributions.MixtureModel method)
(pixyz.distributions.MultiplyDistribution method)
(pixyz.distributions.ProductOfNormal method)
(pixyz.distributions.ReplaceVarDistribution method)
(pixyz.distributions.distributions.Distribution method)
(pixyz.flows.Flow method)
(pixyz.flows.FlowList method)
(pixyz.losses.losses.Loss method)
(pixyz.models.GAN method)
(pixyz.models.ML method)
(pixyz.models.Model method)
(pixyz.models.VAE method)
(pixyz.models.VI method)
A
abs() (pixyz.losses.losses.Loss method)
AbsLoss (class in pixyz.losses.losses)
ActNorm2d (class in pixyz.flows)
AddLoss (class in pixyz.losses.losses)
AdversarialJensenShannon (class in pixyz.losses)
AdversarialKullbackLeibler (class in pixyz.losses)
AdversarialWassersteinDistance (class in pixyz.losses)
AffineCoupling (class in pixyz.flows)
B
BatchMean (class in pixyz.losses.losses)
BatchNorm1d (class in pixyz.flows)
BatchNorm2d (class in pixyz.flows)
BatchSum (class in pixyz.losses.losses)
Bernoulli (class in pixyz.distributions)
Beta (class in pixyz.distributions)
build_mask() (pixyz.flows.AffineCoupling method)
C
Categorical (class in pixyz.distributions)
ChannelConv (class in pixyz.flows)
cond_var (pixyz.distributions.distributions.Distribution attribute)
constant_var() (pixyz.losses.losses.Loss method)
ConstantVar (class in pixyz.losses.losses)
convert_latex_name() (in module pixyz.utils)
CrossEntropy() (in module pixyz.losses)
CustomProb (class in pixyz.distributions)
D
d_loss() (pixyz.losses.AdversarialJensenShannon method)
(pixyz.losses.AdversarialKullbackLeibler method)
(pixyz.losses.AdversarialWassersteinDistance method)
DataParalleledLoss (class in pixyz.losses.losses)
delete_dict_values() (in module pixyz.utils)
deriv_tanh() (pixyz.flows.PlanarFlow method)
Detach (class in pixyz.losses.losses)
detach() (pixyz.losses.losses.Loss method)
detach_dict() (in module pixyz.utils)
Deterministic (class in pixyz.distributions)
Dirichlet (class in pixyz.distributions)
Distribution (class in pixyz.distributions.distributions)
distribution_name (pixyz.distributions.Bernoulli attribute)
(pixyz.distributions.Beta attribute)
(pixyz.distributions.Categorical attribute)
(pixyz.distributions.CustomProb attribute)
(pixyz.distributions.Deterministic attribute)
(pixyz.distributions.Dirichlet attribute)
(pixyz.distributions.EmpiricalDistribution attribute)
(pixyz.distributions.FactorizedBernoulli attribute)
(pixyz.distributions.Gamma attribute)
(pixyz.distributions.InverseTransformedDistribution attribute)
(pixyz.distributions.Laplace attribute)
(pixyz.distributions.MarginalizeVarDistribution attribute)
(pixyz.distributions.MixtureModel attribute)
(pixyz.distributions.Normal attribute)
(pixyz.distributions.RelaxedBernoulli attribute)
(pixyz.distributions.RelaxedCategorical attribute)
(pixyz.distributions.ReplaceVarDistribution attribute)
(pixyz.distributions.TransformedDistribution attribute)
(pixyz.distributions.distributions.Distribution attribute)
distribution_torch_class (pixyz.distributions.Bernoulli attribute)
(pixyz.distributions.Beta attribute)
(pixyz.distributions.Categorical attribute)
(pixyz.distributions.Dirichlet attribute)
(pixyz.distributions.Gamma attribute)
(pixyz.distributions.Laplace attribute)
(pixyz.distributions.Normal attribute)
(pixyz.distributions.RelaxedBernoulli attribute)
(pixyz.distributions.RelaxedCategorical attribute)
DivLoss (class in pixyz.losses.losses)
E
ELBO() (in module pixyz.losses)
ElementWiseProductOfNormal (class in pixyz.distributions)
EmpiricalDistribution (class in pixyz.distributions)
Entropy() (in module pixyz.losses)
epsilon() (in module pixyz.utils)
eval() (pixyz.losses.losses.Loss method)
Expectation (class in pixyz.losses)
expectation() (pixyz.losses.losses.Loss method)
extra_repr() (pixyz.distributions.distributions.Distribution method)
(pixyz.flows.AffineCoupling method)
(pixyz.flows.PlanarFlow method)
F
FactorizedBernoulli (class in pixyz.distributions)
features_shape (pixyz.distributions.distributions.Distribution attribute)
Flatten (class in pixyz.flows)
Flow (class in pixyz.flows)
flow_input_var (pixyz.distributions.TransformedDistribution attribute)
flow_output_var (pixyz.distributions.InverseTransformedDistribution attribute)
FlowList (class in pixyz.flows)
forward() (pixyz.distributions.distributions.Distribution method)
(pixyz.distributions.InverseTransformedDistribution method)
(pixyz.distributions.MarginalizeVarDistribution method)
(pixyz.distributions.ReplaceVarDistribution method)
(pixyz.distributions.TransformedDistribution method)
(pixyz.flows.ActNorm2d method)
(pixyz.flows.AffineCoupling method)
(pixyz.flows.BatchNorm1d method)
(pixyz.flows.ChannelConv method)
(pixyz.flows.Flatten method)
(pixyz.flows.Flow method)
(pixyz.flows.FlowList method)
(pixyz.flows.Permutation method)
(pixyz.flows.PlanarFlow method)
(pixyz.flows.Preprocess method)
(pixyz.flows.Squeeze method)
(pixyz.flows.Unsqueeze method)
(pixyz.losses.AdversarialJensenShannon method)
(pixyz.losses.AdversarialKullbackLeibler method)
(pixyz.losses.Expectation method)
(pixyz.losses.IterativeLoss method)
(pixyz.losses.LogProb method)
(pixyz.losses.MMD method)
(pixyz.losses.Prob method)
(pixyz.losses.WassersteinDistance method)
(pixyz.losses.losses.AbsLoss method)
(pixyz.losses.losses.AddLoss method)
(pixyz.losses.losses.BatchMean method)
(pixyz.losses.losses.BatchSum method)
(pixyz.losses.losses.ConstantVar method)
(pixyz.losses.losses.DataParalleledLoss method)
(pixyz.losses.losses.Detach method)
(pixyz.losses.losses.DivLoss method)
(pixyz.losses.losses.Loss method)
(pixyz.losses.losses.LossOperator method)
(pixyz.losses.losses.MaxLoss method)
(pixyz.losses.losses.MinLoss method)
(pixyz.losses.losses.MulLoss method)
(pixyz.losses.losses.NegLoss method)
(pixyz.losses.losses.Parameter method)
(pixyz.losses.losses.SubLoss method)
(pixyz.losses.losses.ValueLoss method)
FrozenSampleDict (class in pixyz.utils)
G
g_loss() (pixyz.losses.AdversarialJensenShannon method)
(pixyz.losses.AdversarialKullbackLeibler method)
(pixyz.losses.AdversarialWassersteinDistance method)
Gamma (class in pixyz.distributions)
GAN (class in pixyz.models)
get_dict_values() (in module pixyz.utils)
get_entropy() (pixyz.distributions.distributions.Distribution method)
(pixyz.distributions.MarginalizeVarDistribution method)
(pixyz.distributions.ReplaceVarDistribution method)
get_log_prob() (pixyz.distributions.CustomProb method)
(pixyz.distributions.Deterministic method)
(pixyz.distributions.EmpiricalDistribution method)
(pixyz.distributions.FactorizedBernoulli method)
(pixyz.distributions.InverseTransformedDistribution method)
(pixyz.distributions.MixtureModel method)
(pixyz.distributions.ProductOfNormal method)
(pixyz.distributions.TransformedDistribution method)
(pixyz.distributions.distributions.Distribution method)
get_parameters() (pixyz.flows.AffineCoupling method)
(pixyz.flows.ChannelConv method)
get_params() (pixyz.distributions.ProductOfNormal method)
graph (pixyz.distributions.distributions.Distribution attribute)
H
has_reparam (pixyz.distributions.Bernoulli attribute)
(pixyz.distributions.Beta attribute)
(pixyz.distributions.Categorical attribute)
(pixyz.distributions.CustomProb attribute)
(pixyz.distributions.Deterministic attribute)
(pixyz.distributions.Dirichlet attribute)
(pixyz.distributions.EmpiricalDistribution attribute)
(pixyz.distributions.Gamma attribute)
(pixyz.distributions.InverseTransformedDistribution attribute)
(pixyz.distributions.Laplace attribute)
(pixyz.distributions.MixtureModel attribute)
(pixyz.distributions.Normal attribute)
(pixyz.distributions.RelaxedBernoulli attribute)
(pixyz.distributions.RelaxedCategorical attribute)
(pixyz.distributions.TransformedDistribution attribute)
(pixyz.distributions.distributions.Distribution attribute)
hidden_var (pixyz.distributions.MixtureModel attribute)
I
in_features (pixyz.flows.Flow attribute)
inference() (pixyz.distributions.InverseTransformedDistribution method)
initialize_parameters() (pixyz.flows.ActNorm2d method)
input_var (pixyz.distributions.CustomProb attribute)
(pixyz.distributions.EmpiricalDistribution attribute)
(pixyz.distributions.distributions.Distribution attribute)
(pixyz.losses.losses.Loss attribute)
inverse() (pixyz.distributions.InverseTransformedDistribution method)
(pixyz.distributions.TransformedDistribution method)
(pixyz.flows.ActNorm2d method)
(pixyz.flows.AffineCoupling method)
(pixyz.flows.BatchNorm1d method)
(pixyz.flows.ChannelConv method)
(pixyz.flows.Flatten method)
(pixyz.flows.Flow method)
(pixyz.flows.FlowList method)
(pixyz.flows.Permutation method)
(pixyz.flows.PlanarFlow method)
(pixyz.flows.Preprocess method)
(pixyz.flows.Squeeze method)
(pixyz.flows.Unsqueeze method)
InverseTransformedDistribution (class in pixyz.distributions)
IterativeLoss (class in pixyz.losses)
K
KullbackLeibler() (in module pixyz.losses)
L
Laplace (class in pixyz.distributions)
load() (pixyz.models.Model method)
log_prob() (pixyz.distributions.distributions.Distribution method)
(pixyz.distributions.ProductOfNormal method)
log_prob_function (pixyz.distributions.CustomProb attribute)
logdet_jacobian (pixyz.distributions.InverseTransformedDistribution attribute)
(pixyz.distributions.TransformedDistribution attribute)
(pixyz.flows.Flow attribute)
logit() (pixyz.flows.Preprocess static method)
LogProb (class in pixyz.losses)
Loss (class in pixyz.losses.losses)
loss_test() (pixyz.losses.AdversarialJensenShannon method)
(pixyz.losses.AdversarialKullbackLeibler method)
(pixyz.losses.AdversarialWassersteinDistance method)
(pixyz.losses.losses.LossSelfOperator method)
loss_text (pixyz.losses.losses.Loss attribute)
loss_train() (pixyz.losses.AdversarialJensenShannon method)
(pixyz.losses.AdversarialKullbackLeibler method)
(pixyz.losses.AdversarialWassersteinDistance method)
(pixyz.losses.losses.LossSelfOperator method)
LossOperator (class in pixyz.losses.losses)
LossSelfOperator (class in pixyz.losses.losses)
lru_cache_for_sample_dict() (in module pixyz.utils)
M
marginalize_var() (pixyz.distributions.distributions.Distribution method)
MarginalizeVarDistribution (class in pixyz.distributions)
MaxLoss (class in pixyz.losses.losses)
mean() (pixyz.losses.losses.Loss method)
MinLoss (class in pixyz.losses.losses)
MixtureModel (class in pixyz.distributions)
ML (class in pixyz.models)
MMD (class in pixyz.losses)
Model (class in pixyz.models)
MulLoss (class in pixyz.losses.losses)
MultiplyDistribution (class in pixyz.distributions)
N
name (pixyz.distributions.distributions.Distribution attribute)
NegLoss (class in pixyz.losses.losses)
Normal (class in pixyz.distributions)
P
Parameter (class in pixyz.losses.losses)
params_keys (pixyz.distributions.Bernoulli attribute)
(pixyz.distributions.Beta attribute)
(pixyz.distributions.Categorical attribute)
(pixyz.distributions.Dirichlet attribute)
(pixyz.distributions.Gamma attribute)
(pixyz.distributions.Laplace attribute)
(pixyz.distributions.Normal attribute)
(pixyz.distributions.RelaxedBernoulli attribute)
(pixyz.distributions.RelaxedCategorical attribute)
Permutation (class in pixyz.flows)
pixyz.distributions (module)
pixyz.flows (module)
pixyz.losses (module)
pixyz.models (module)
pixyz.utils (module)
PlanarFlow (class in pixyz.flows)
posterior() (pixyz.distributions.MixtureModel method)
Preprocess (class in pixyz.flows)
print_latex() (in module pixyz.utils)
Prob (class in pixyz.losses)
prob() (pixyz.distributions.distributions.Distribution method)
(pixyz.distributions.ProductOfNormal method)
prob_factorized_text (pixyz.distributions.distributions.Distribution attribute)
(pixyz.distributions.InverseTransformedDistribution attribute)
(pixyz.distributions.MixtureModel attribute)
(pixyz.distributions.ProductOfNormal attribute)
(pixyz.distributions.TransformedDistribution attribute)
prob_joint_factorized_and_text (pixyz.distributions.distributions.Distribution attribute)
(pixyz.distributions.ProductOfNormal attribute)
prob_text (pixyz.distributions.distributions.Distribution attribute)
ProductOfNormal (class in pixyz.distributions)
R
REINFORCE() (in module pixyz.losses)
RelaxedBernoulli (class in pixyz.distributions)
RelaxedCategorical (class in pixyz.distributions)
replace_dict_keys() (in module pixyz.utils)
replace_dict_keys_split() (in module pixyz.utils)
replace_var() (pixyz.distributions.distributions.Distribution method)
ReplaceVarDistribution (class in pixyz.distributions)
reset_parameters() (pixyz.flows.PlanarFlow method)
Reverse (class in pixyz.flows)
S
sample() (pixyz.distributions.CustomProb method)
(pixyz.distributions.Deterministic method)
(pixyz.distributions.EmpiricalDistribution method)
(pixyz.distributions.InverseTransformedDistribution method)
(pixyz.distributions.MixtureModel method)
(pixyz.distributions.RelaxedBernoulli method)
(pixyz.distributions.RelaxedCategorical method)
(pixyz.distributions.TransformedDistribution method)
(pixyz.distributions.distributions.Distribution method)
sample_mean() (pixyz.distributions.Deterministic method)
(pixyz.distributions.EmpiricalDistribution method)
(pixyz.distributions.MarginalizeVarDistribution method)
(pixyz.distributions.ReplaceVarDistribution method)
(pixyz.distributions.distributions.Distribution method)
sample_variance() (pixyz.distributions.distributions.Distribution method)
(pixyz.distributions.MarginalizeVarDistribution method)
(pixyz.distributions.ReplaceVarDistribution method)
save() (pixyz.models.Model method)
set_dist() (pixyz.distributions.RelaxedBernoulli method)
(pixyz.distributions.RelaxedCategorical method)
set_epsilon() (in module pixyz.utils)
set_loss() (pixyz.models.Model method)
Shuffle (class in pixyz.flows)
slice_step_fn() (pixyz.losses.IterativeLoss method)
Squeeze (class in pixyz.flows)
SubLoss (class in pixyz.losses.losses)
sum() (pixyz.losses.losses.Loss method)
sum_samples() (in module pixyz.utils)
T
test() (pixyz.models.GAN method)
(pixyz.models.ML method)
(pixyz.models.Model method)
(pixyz.models.VAE method)
(pixyz.models.VI method)
tolist() (in module pixyz.utils)
train() (pixyz.models.GAN method)
(pixyz.models.ML method)
(pixyz.models.Model method)
(pixyz.models.VAE method)
(pixyz.models.VI method)
TransformedDistribution (class in pixyz.distributions)
U
Unsqueeze (class in pixyz.flows)
V
VAE (class in pixyz.models)
ValueLoss (class in pixyz.losses.losses)
var (pixyz.distributions.distributions.Distribution attribute)
VI (class in pixyz.models)
W
WassersteinDistance (class in pixyz.losses)