如何在tensorflow中构建具有自定义激活函数的神经网络？

import tensorflow as tf

def custom_activation(x):
    return tf.nn.relu(tf.sin(x))

model = tf.keras.models.Sequential([
    tf.keras.layers.Dense(64, activation=custom_activation, input_shape=(input_shape,)),
    tf.keras.layers.Dense(10, activation='softmax')
])

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

model.fit(x_train, y_train, epochs=10, batch_size=32)