def expected_token_logproba(self) -> tf.Tensor:
        """Logarithmic probability to predict label token.

        Returns:shape = [batch_size, max_logit_length, max_label_length + 1]
        """
        label_logproba = tf.gather(
            params=self.logproba,
            indices=self.label,
            axis=2,
            batch_dims=1,
        )
        expected_token_logproba = apply_logarithmic_mask(label_logproba, tf.expand_dims(self.label_length_mask, axis=1))
        # shape = [batch_size, max_logit_length, max_label_length + 1]
        return expected_token_logproba
Logarithmic probability to predict label token.

        Returns:shape = [batch_size, max_logit_length, max_label_length + 1]
        
tensorflow_asr/losses/impl/ctc_tpu.py