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)[源代码]