Expand description
Tensor math used in optimizer algorithms
Implements functions that burn_tensor does not provide.
Functionsยง
- digamma_
tensor - Approximate derivative of the log gamma function (digamma)
- ln_
gamma_ tensor - Approximate log-gamma function
- logsumexp
- LogSumExp over a dimension on a 2D tensor
- logsumexp_
mask - LogSumExp on a masked 2D tensor
- logsumexp_
mat - LogSumExp on a 2D tensor