Dense ResNet Like Layer

dense_resnet.py

class DenseResNet(tf.keras.layers.Layer):

  def __init__(self, units=500, activation="tanh", kernel_initializer="glorot_uniform", layers=2):
    super(DenseResNet, self).__init__()
    self.layers = [tf.keras.layers.Dense(units, activation=activation, kernel_initializer=kernel_initializer) for _ in range(layers)]

  def call(self, inputs):
    result = inputs
    for layer in self.layers:
      result = layer(result)
    return tf.keras.layers.Add()([inputs,result])