Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.nn.collapse_repeated

TensorFlow 2.0 version View source on GitHub

Merge repeated labels into single labels.

Aliases:

  • tf.compat.v1.nn.collapse_repeated
  • tf.compat.v2.nn.collapse_repeated
tf.nn.collapse_repeated(
    labels,
    seq_length,
    name=None
)

Args:

  • labels: Tensor of shape [batch, max value in seq_length]
  • seq_length: Tensor of shape [batch], sequence length of each batch element.
  • name: A name for this Op. Defaults to "collapse_repeated_labels".

Returns:

A tuple (collapsed_labels, new_seq_length) where

  • collapsed_labels: Tensor of shape [batch, max_seq_length] with repeated labels collapsed and padded to max_seq_length, eg: [[A, A, B, B, A], [A, B, C, D, E]] => [[A, B, A, 0, 0], [A, B, C, D, E]]

  • new_seq_length: int tensor of shape [batch] with new sequence lengths.