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);