Merge repeated labels into single labels.
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.
|