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])