mindspore.mint.logsumexp

mindspore.mint.logsumexp(input, dim, keepdim=False)

计算tensor在指定维度上的指数和的对数。

\[logsumexp(input) = \log(\sum(e^{input-input_{max}})) + input_{max}\]

警告

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

参数：

• input (Tensor) - 输入tensor。

• dim (Union[int, tuple(int), list(int)]) - 指定维度。如果为 () ，计算 input 中的所有元素。

• keepdim (bool，可选) - 输出tensor是否保留维度。默认 False。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import numpy as np
>>> import mindspore
>>> x = mindspore.tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
>>> output = mindspore.mint.logsumexp(x, 1, keepdim=True)
>>> print(output.shape)
(3, 1, 5, 6)