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.GammaDistribution(concentration, rate)

For now only support gamma distribution.

cdf(value)

log_prob(value)

mean()

mode()

sample(shape)

variance()

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)