distr_gamma: Creates a Gamma distribution parameterized by shape...

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Creates a Gamma distribution parameterized by shape concentration and rate.

Description

Creates a Gamma distribution parameterized by shape concentration and rate.

Usage

distr_gamma(concentration, rate, validate_args = NULL)

Arguments

concentration	(float or Tensor): shape parameter of the distribution (often referred to as alpha)
rate	(float or Tensor): rate = 1 / scale of the distribution (often referred to as beta)
validate_args	whether to validate arguments or not.

See Also

Distribution for details on the available methods.

Other distributions: distr_bernoulli(), distr_chi2(), distr_multivariate_normal(), distr_normal(), distr_poisson()

Examples

if (torch_is_installed()) {
m <- distr_gamma(torch_tensor(1.0), torch_tensor(1.0))
m$sample() # Gamma distributed with concentration=1 and rate=1
}

torch documentation built on May 29, 2024, 9:54 a.m.