Commit 5d93b995 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Derivative of tf.random_gamma with respect to the alpha parameter. 

Previously, tf.random_gamma(shape, alpha, beta) was differentiable only w.r.t. beta. This commit adds the derivative w.r.t. alpha. The implementation is based on Eigen's gamma_sample_der_alpha function, which computes the "implicit reparameterization" derivative. This function is not directly exposed in the public TensorFlow API.

PiperOrigin-RevId: 201258617
parent a4553192
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment