mindspore.mint.nn.LogSigmoid

查看源文件
class mindspore.mint.nn.LogSigmoid[源代码]

逐元素计算LogSigmoid激活函数。输入是任意格式的Tensor。

LogSigmoid定义为:

\[\text{LogSigmoid}(x_{i}) = \log(\frac{1}{1 + \exp(-x_i)}),\]

其中,\(x_{i}\) 是输入Tensor的一个元素。

LogSigmoid函数图:

../../_images/LogSigmoid.png

警告

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

输入:
  • input (Tensor) - LogSigmoid的输入,数据类型为bfloat16,float16或float32。shape为 \((*)\) ,其中 \(*\) 表示任意的附加维度。

输出:

Tensor,数据类型和shape与 input 的相同。

异常:
  • TypeError - input 的数据类型既不是bfloat16,float16也不是float32。

  • TypeError - input 不是一个Tensor。

支持平台:

Ascend

样例:

>>> import mindspore
>>> from mindspore import Tensor
>>> net = mint.nn.LogSigmoid()
>>> input = Tensor([1.0, 2.0, 3.0], mindspore.float32)
>>> output = net(input)
>>> print(output)
[-0.31326166 -0.12692806 -0.04858734]