Python PyTorch VonMises用法及代碼示例

本文簡要介紹python語言中 torch.distributions.von_mises.VonMises 的用法。

用法:

class torch.distributions.von_mises.VonMises(loc, concentration, validate_args=None)

參數：

• loc(torch.Tensor) -以弧度為單位的角度。

• concentration(torch.Tensor) -濃度參數

基礎：torch.distributions.distribution.Distribution

圓形 von Mises 分布。

此實現使用極坐標。 loc 和 value 參數可以是任何實數(以促進無約束優化)，但被解釋為模 2 pi 的角度。

例子：

>>> m = dist.VonMises(torch.tensor([1.0]), torch.tensor([1.0]))
>>> m.sample() # von Mises distributed with loc=1 and concentration=1
tensor([1.9777])