rockpool.training.ctc_loss

Implementations of the CTC loss, in numpy, Jax and Torch

Functions

ctc_loss_jax(label, log_prob, seq_length[, ...])

Jax-compatible implementation of the CTC loss

ctc_loss_numpy(label, log_prob, seq_length)

Numpy implementation of the CTC loss

ctc_loss_torch(label, log_prob, seq_length)

Torch implementation of the CTC loss