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

TensorLike(...): Union type; Union[X, Y] means either X or Y.

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