diff --git a/docs/reference/ptd_gamma.md b/docs/reference/ptd_gamma.md index 1dbcb3d..198150e 100644 --- a/docs/reference/ptd_gamma.md +++ b/docs/reference/ptd_gamma.md @@ -2,8 +2,8 @@ Instantiates a Gamma distribution object. Args: - alpha: Rate. Mean of the distsribution. You can specify a scalar or a tuple of float. - beta: Rate. Standard deviation of the distribution. You can specify a scalar or a tuple of float. + alpha: Shape. You can specify a scalar or a tuple of float. + beta: Rate. You can specify a scalar or a tuple of float. ## Input | Name | Data type | diff --git a/modules/pytorch_wrapper/ptd_gamma.py b/modules/pytorch_wrapper/ptd_gamma.py index 14080fe..f2e4fbe 100644 --- a/modules/pytorch_wrapper/ptd_gamma.py +++ b/modules/pytorch_wrapper/ptd_gamma.py @@ -8,8 +8,8 @@ class PtdGamma: Instantiates a Gamma distribution object. Args: - alpha: Rate. Mean of the distsribution. You can specify a scalar or a tuple of float. - beta: Rate. Standard deviation of the distribution. You can specify a scalar or a tuple of float. + alpha: Shape. You can specify a scalar or a tuple of float. + beta: Rate. You can specify a scalar or a tuple of float. category: PyTorch wrapper - Distribution """