Function torch::special::logsumexp

• Defined in File special.h

Function Documentation

inline Tensor torch::special::logsumexp(const Tensor &self, IntArrayRef dims, bool keepdim)

Computes the log of summed exponentials of each row of input in the given dimension dim See https://pytorch.org/docs/main/special.html#torch.special.logsumexp.

Example:

auto t = torch::randn(3, 3);
torch::special::logsumexp(t, 1);