tfa.seq2seq.sequence_loss | TensorFlow Addons
www.tensorflow.org › tfa › seq2seqNov 15, 2021 · Computes the weighted cross-entropy loss for a sequence of logits. Depending on the values of average_across_timesteps / sum_over_timesteps and average_across_batch / sum_over_batch, the return Tensor will have rank 0, 1, or 2 as these arguments reduce the cross-entropy at each target, which has shape [batch_size, sequence_length], over their ...