Python PyTorch Chi2用法及代码示例

本文简要介绍python语言中 torch.distributions.chi2.Chi2 的用法。

用法:

class torch.distributions.chi2.Chi2(df, validate_args=None)

参数：

df(float或者Tensor) -分布的形状参数

基础：torch.distributions.gamma.Gamma

创建由形状参数 df 参数化的 Chi2 分布。这完全相当于Gamma(alpha=0.5*df, beta=0.5)

例子：

>>> m = Chi2(torch.tensor([1.0]))
>>> m.sample()  # Chi2 distributed with shape df=1
tensor([ 0.1046])