def call(self, inputs, **kwargs):
    stacked = self.stack_features(inputs)
    interactions = self.interaction_layer(stacked)
    return (interactions if (not self.top_mlp) else self.top_mlp(interactions))
def call(self, inputs, **kwargs):
    stacked = self.stack_features(inputs)
    interactions = self.interaction_layer(stacked)
    return (interactions if (not self.top_mlp) else self.top_mlp(interactions))