cupyx.scipy.special.logsumexp#

cupyx.scipy.special.logsumexp(a, axis=None, b=None, keepdims=False, return_sign=False)[源代码]#

计算输入元素指数和的对数。

参数:

  • a (cupy.ndarray) – 输入数组

  • axis (None or int or tuple of ints, optional) – 执行求和运算的轴。默认为 None，表示对所有元素求和

  • keepdims (bool, optional) – 如果设置为 True，则减小的轴将作为大小为 1 的维度保留在结果中。使用此选项，结果将与原始数组正确广播

  • b (cupy.ndarray, optional) – exp(a) 的缩放因子，必须与 a 的形状相同或可广播到 a。这些值可以为负以实现减法

  • return_sign (bool, optional) – 如果设置为 True，结果将是包含符号信息的对；如果为 False，则负数结果将作为 NaN 返回。默认为 False

返回:

  • res (cupy.ndarray) – 结果，以数值更稳定的方式计算的 cp.log(cp.sum(cp.exp(a)))。如果给定 b，则返回 cp.log(cp.sum(b*cp.exp(a)))

  • sgn (cupy.ndarray) – 如果 return_sign 为 True，这将是一个与 res 匹配的浮点数数组，并根据结果的符号为 +1、0 或 -1。如果为 False，则只返回一个结果。

另请参阅

scipy.special.logsumexp