I have a PyTorch model that I am attempting to rewrite in Keras which I have minimal experience with. This is the PyTorch model
class Model(nn.Module): def __