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