Module: tfa.seq2seq.loss

View source on GitHub

Seq2seq loss operations for use in sequence models.

Classes

class SequenceLoss: Weighted cross-entropy loss for a sequence of logits.

Functions

sequence_loss(...): Weighted cross-entropy loss for a sequence of logits.