Function dfdx::tensor_ops::logsumexp

source · [−]
pub fn logsumexp<T: Reduce<Axes>, Axes>(t: T) -> T::Reduced
Expand description

Computes the LogSumExp function across Axes

Pytorch equivalent: t.exp().sum(Axes).log()

Related functions: ln(), sum(), exp(), log_softmax(), softmax()

Example:

let t: Tensor3D<2, 4, 6> = TensorCreator::zeros();
let _: Tensor2D<2, 4> = t.logsumexp();

Multi axis logsumexp:

let _: Tensor1D<4> = t.logsumexp();