INNER CODE UNIT · Python

mish_auto

digantamisra98/Mish · PyTorch Benchmarks/activations_autofn.py:60

def mish_auto(x, inplace=False):
    # inplace ignored
    return MishAutoFn.apply(x)


class MishAuto(nn.Module):
    def __init__(self, inplace: bool = False):
        super(MishAuto, self).__init__()
        self.inplace = inplace

    def forward(self, x):
        return MishAutoFn.apply(x)

View source record →