src/transformers/models/hubert/modeling_tf_hubert.py/TFHubertWeightNormConv1D/_init_norm
class TFHubertWeightNormConv1D:    def _init_norm(self):
        """Set the norm of the weight vector."""
        kernel_norm = tf.sqrt(tf.reduce_sum(tf.square(self.weight_v), axis=self.kernel_norm_axes))
        self.weight_g.assign(kernel_norm[:, tf.newaxis, tf.newaxis])