pytorch / 2 / generated / torch.nn.functional.celu.html

torch.nn.functional.celu

torch.nn.functional.celu(input, alpha=1., inplace=False) → Tensor [source]

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

See CELU for more details.

Return type

Tensor

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.functional.celu.html