tf.keras.ops.ctc_loss

CTC (Connectionist Temporal Classification) loss.

View aliases

Main aliases

tf.keras.ops.nn.ctc_loss

tf.keras.ops.ctc_loss(
    target, output, target_length, output_length, mask_index=0
)

Args
target	A tensor of shape (batch_size, max_length) containing the true labels in integer format.
output	A tensor of shape (batch_size, max_length, num_classes) containing logits (the output of your model).
target_length	A tensor of shape (batch_size,) containing the true label lengths.
output_length	A tensor of shape (batch_size,) containing the output lengths.
mask_index	The index of the mask character in the vocabulary. Defaults to 0.