tensorflow Smish version

[tomaszjacek]

Mr. Professor,

When I'm trying

---

@tf.function
def smish(input):
return input * tf.math.tanh(tf.math.log(1+tf.math.sigmoid(input)))

class Smish(layers.Layer):

def __init__(self, input):
    """
    Init method.
    """
    super().__init__()

def call(self, input):
    """
    Forward pass of the function.
    """
    return smish(input)

---

than in model declaration

self_conv2D_11 = Conv2D(16 , kernel_size=(3,3),strides=(1,1),padding = 'same')

activation_9 = Smish(conv2D_1)
conv2D_11 = self_conv2D_11(activation_9)

i got an error
TypeError: Inputs to a layer should be tensors. Got: <main.Smish object at 0x000001F82C4FE910>

Would you be so kind to give hint how to translate Smish to tensorflow ?

Thank you,
tj

[tomaszjacek]

it works, just need to change
def init(self, input):
to
def init(self):
and declare first
self_activation_9 = Smish()
than use it as
activation_9 = self_activation_9(conv2D_1)