View source on GitHub |
Get unique labels and indices for batched labels for tf.nn.ctc_loss
.
tf.nn.ctc_unique_labels( labels, name=None )
For use with tf.nn.ctc_loss
optional argument unique
: This op can be used to preprocess labels in input pipeline to for better speed/memory use computing the ctc loss on TPU.
ctc_unique_labels([[3, 4, 4, 3]]) -> unique labels padded with 0: [[3, 4, 0, 0]] indices of original labels in unique: [0, 1, 1, 0]
Args | |
---|---|
labels | tensor of shape [batch_size, max_label_length] padded with 0. |
name | A name for this Op . Defaults to "ctc_unique_labels". |
Returns | |
---|---|
tuple of
|
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/nn/ctc_unique_labels