mindspore.ops.CeLU

View Source On Gitee
class mindspore.ops.CeLU(alpha=1.0)[source]

Computes CeLU (Continuously differentiable exponential linear units) of input tensors element-wise.

Refer to mindspore.ops.celu() for more details.

Warning

This is an experimental API that is subject to change or deletion.

Parameters

alpha (float, optional) – The \(\alpha\) value for the Celu formulation. Default: 1.0 .

Inputs:
  • input_x (Tensor) - The input tensor with a dtype of float16 or float32.

Outputs:

Tensor, with the same type and shape as the input_x.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([-2.0, -1.0, 1.0, 2.0]), mindspore.float32)
>>> celu = ops.CeLU(alpha=1.0)
>>> output = celu(input_x)
>>> print(output)
[-0.86466473 -0.63212055  1.          2.        ]
>>> input_x = Tensor(2.1, mindspore.float32)
>>> output = celu(input_x)
>>> print(output)
2.1