rtkclouds

class Rezero(layers.Layer):
    def __init__(self):
        super().__init__()
        self.alpha1 = tf.Variable(0.0, trainable=True)

    def call(self, inputs, training):
        return self.alpha1*inputs
class CustomRezero(tf.keras.layers.Layer):