pymc.math.logsumexp#

pymc.math.logsumexp(x,axis=None,keepdims=False)[source]#

Compute the log of the sum of exponentials of input elements.

Seescipy.special.logsumexp.

Parameters:
xsymbolictensor

Input

axisNone orint ortuple ofints, optional

Axis or axes over which the sum is taken. By default axis is None,and all elements are summed.

keepdimsbool, optional

If this is set to True, the axes which are reduced are left in theresult as dimensions with size one. With this option, the result willbroadcast correctly against the original array.

Returns:
TensorVariable