I am creating a custom model, where MyLayer is shared and applied to different inputs:
MyLayer
class MyModel(keras.models.Model): def __init__(self)