Source code for cvnets.layers.activation.tanh

#
# For licensing see accompanying LICENSE file.
# Copyright (C) 2023 Apple Inc. All Rights Reserved.
#

from torch import Tensor, nn

from cvnets.layers.activation import register_act_fn


@register_act_fn(name="tanh")
class Tanh(nn.Tanh):
    """
    Applies Tanh function
    """

    def __init__(self, *args, **kwargs) -> None:
        super().__init__()