OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Clarify meaning of rate parameter in Gamma distribution (#134847)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134847
Approved by: https://github.com/fritzo
This commit is contained in:
Peter Steinbach 2024-11-09 00:22:13 +00:00 committed by PyTorch MergeBot
parent 7eb66173e2
commit 090b778b8a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributions/gamma.py
View File

@ -29,8 +29,8 @@ class Gamma(ExponentialFamily):
Args: Args:
concentration (float or Tensor): shape parameter of the distribution concentration (float or Tensor): shape parameter of the distribution
(often referred to as alpha) (often referred to as alpha)
rate (float or Tensor): rate = 1 / scale of the distribution rate (float or Tensor): rate parameter of the distribution
(often referred to as beta) (often referred to as beta), rate = 1 / scale
""" """
arg_constraints = { arg_constraints = {
"concentration": constraints.positive, "concentration": constraints.positive,