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 ]