pytorch / 2 / generated / torch.nn.mish.html

Mish

class torch.nn.Mish(inplace=False) [source]

Applies the Mish function, element-wise. Mish: A Self Regularized Non-Monotonic Neural Activation Function.

Mish ( x ) = x Tanh ( Softplus ( x ) ) \text{Mish}(x) = x * \text{Tanh}(\text{Softplus}(x))
Shape:
  • Input: ( ) (*) , where * means any number of dimensions.
  • Output: ( ) (*) , same shape as the input.
../_images/Mish.png

Examples:

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

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