jittor.distributions
这里是Jittor的随机分布模块的API文档,您可以通过from jittor import distributions
来获取该模块。
-
class jittor.distributions.Categorical(probs=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=())[源代码]
-
class jittor.distributions.Geometric(p=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape)[源代码]
-
class jittor.distributions.Normal(mu, sigma)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=None)[源代码]
-
class jittor.distributions.OneHotCategorical(probs=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=[])[源代码]
-
class jittor.distributions.Uniform(low, high)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape)[源代码]
-
jittor.distributions.kl_divergence(cur_dist, old_dist)[源代码]
-
jittor.distributions.simple_presum(x)[源代码]