class MyLayer(tf.keras.layers.Layer):
  def __init__(self):
  super().__init__()
  def build(self, input_shape):
    pass
  def call(self, input):
    pass
  def compute_output_shape(self, input_shape):
    pass