mindspore.mint.logaddexp2

查看源文件
mindspore.mint.logaddexp2(input, other)[源代码]

计算以2为底的输入的指数和的对数。

\[out_i = \log_2(2^{input_i} + 2^{other_i})\]
参数:
  • input (Tensor) - 输入Tensor,其数据类型必须是float。

  • other (Tensor) - 输入Tensor,其数据类型必须是float。如果 input 的shape不等于 other 的shape,它们必须被广播成相同shape(输出的形状)。

返回:

Tensor,数据类型与输入 inputother 相同。

异常:
  • TypeError - inputother 不是Tensor。

  • TypeError - inputother 的数据类型不是float。

支持平台:

Ascend

样例:

>>> import numpy as np
>>> from mindspore import Tensor, mint
>>> x1 = Tensor(np.array([1, 2, 3]).astype(np.float16))
>>> x2 = Tensor(np.array(2).astype(np.float16))
>>> output = mint.logaddexp2(x1, x2)
>>> print(output)
[2.312 2.693 3.312]