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