考虑以下来自 TensorFlow 教程的自定义层代码:
class MyDenseLayer(tf.keras.layers.Layer):
def __init__(self, num_outputs):
super(MyDenseLayer, self).__init__()
self.num_outputs = num_outputs
def build(self, input_shape):
self.kernel = self.add_weight("kernel",
shape=[int(input_shape[-1]),
self.num_outputs])
def call(self, input):
return tf.matmul(input, self.kernel)
如何在自定义层的参数上应用任何预定义的正则化(例如 tf.keras.regularizers.L1
)或自定义正则化?