快捷方式

CELU

class torch.nn.CELU(alpha=1.0, inplace=False)[source]

逐元素应用 CELU 函数。

CELU(x)=max(0,x)+min(0,α(exp(x/α)1))\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x/\alpha) - 1))

可以在论文 连续可微的指数线性单元 中找到更多详细信息。

参数
  • alpha (float) – CELU 公式的 α\alpha 值。默认值:1.0

  • inplace (bool) – 可以选择就地进行操作。默认值:False

形状
  • 输入:()(*),其中 * 表示任意数量的维度。

  • 输出:()(*),与输入相同的形状。

../_images/CELU.png

示例

>>> m = nn.CELU()
>>> input = torch.randn(2)
>>> output = m(input)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源