mindspore.ops.logcumsumexp

mindspore.ops.logcumsumexp(input, axis)

计算输入Tensor input 元素的的指数沿轴 axis 的累积和的对数。例如，如果 input 是 tensor [a, b, c] 并且 axis 是0，返回值为 [a, log(exp(a) + exp(b)), log(exp(a) + exp(b) + exp(c))]。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• input (Tensor) - 任意维度的Tensor。必须是以下几种数据类型：float16、float32、float64。

• axis (int) - 累积计算的轴。必须在 [-rank(input), rank(input)) 的范围之内。

返回：

Tensor，和输入Tensor的dtype和shape相同。

异常：

• TypeError - 如果 input 不是Tensor。

• TypeError - 如果 input 的dtype不在 [float16, float32, float64] 之内。

• TypeError - 如果 axis 不是int。

• ValueError - 如果 axis 超出范围 [-rank(input), rank(input)) 。

支持平台：

Ascend CPU GPU

样例：

>>> import mindspore as ms
>>> from mindspore import ops
>>> import numpy as np
>>> x = ms.Tensor(np.array([1.0, 2.0, 3.0]).astype(np.float32))
>>> output = ops.logcumsumexp(x, 0)
>>> print(output)
[1.        2.3132617 3.407606 ]