def signal_augment(self, inputs, inputs_length):
        """
        Augment audio signals

        Parameters
        ----------
        inputs : tf.Tensor, shape [B, None]
            Original audio signals
        inputs_length : tf.Tensor, shape [B]
            Original audio signals length

        Returns
        -------
        tf.Tensor, shape [B, None]
            Augmented audio signals
        """
        return tf.map_fn(
            fn=lambda x: self._augment(x, self.signal_augmentations),
            elems=(inputs, inputs_length),
            fn_output_signature=(
                tf.TensorSpec.from_tensor(inputs[0]),
                tf.TensorSpec.from_tensor(inputs_length[0]),
            ),
        )

        Augment audio signals

        Parameters
        ----------
        inputs : tf.Tensor, shape [B, None]
            Original audio signals
        inputs_length : tf.Tensor, shape [B]
            Original audio signals length

        Returns
        -------
        tf.Tensor, shape [B, None]
            Augmented audio signals
        
tensorflow_asr/augmentations/augmentation.py