Keras:如何创建自定义嘈杂的Relu函数？

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

class NoisyReLU(layers.Layer):
    def __init__(self, noise_factor=0.1, **kwargs):
        super(NoisyReLU, self).__init__(**kwargs)
        self.noise_factor = noise_factor

    def call(self, inputs):
        noise = tf.random.normal(shape=tf.shape(inputs), mean=0.0, stddev=self.noise_factor)
        noisy_inputs = inputs + noise
        return tf.nn.relu(noisy_inputs)

model = keras.Sequential([
    layers.Dense(64),
    NoisyReLU(),
    layers.Dense(10)
])