mindspore.ops.CeLU
- 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