T
- data type for logProbability()
output@Operator(group="nn") public final class CtcBeamSearchDecoder<T extends Number> extends PrimitiveOp
A note about the attribute merge_repeated: For the beam search decoder, this means that if consecutive entries in a beam are the same, only the first of these is emitted. That is, when the top path is "A B B B B", "A B" is returned if merge_repeated = True but "A B B B B" is returned if merge_repeated = False.
Modifier and Type | Class and Description |
---|---|
static class |
CtcBeamSearchDecoder.Options
Optional attributes for
CtcBeamSearchDecoder |
operation
Modifier and Type | Method and Description |
---|---|
static <T extends Number> |
create(Scope scope,
Operand<T> inputs,
Operand<Integer> sequenceLength,
Long beamWidth,
Long topPaths,
CtcBeamSearchDecoder.Options... options)
Factory method to create a class wrapping a new CtcBeamSearchDecoder operation.
|
List<Output<Long>> |
decodedIndices()
A list (length: top_paths) of indices matrices.
|
List<Output<Long>> |
decodedShape()
A list (length: top_paths) of shape vector.
|
List<Output<Long>> |
decodedValues()
A list (length: top_paths) of values vectors.
|
Output<T> |
logProbability()
A matrix, shaped: `(batch_size x top_paths)`.
|
static CtcBeamSearchDecoder.Options |
mergeRepeated(Boolean mergeRepeated) |
equals, hashCode, op, toString
public static <T extends Number> CtcBeamSearchDecoder<T> create(Scope scope, Operand<T> inputs, Operand<Integer> sequenceLength, Long beamWidth, Long topPaths, CtcBeamSearchDecoder.Options... options)
scope
- current scopeinputs
- 3-D, shape: `(max_time x batch_size x num_classes)`, the logits.sequenceLength
- A vector containing sequence lengths, size `(batch)`.beamWidth
- A scalar >= 0 (beam search beam width).topPaths
- A scalar >= 0, <= beam_width (controls output size).options
- carries optional attributes valuespublic static CtcBeamSearchDecoder.Options mergeRepeated(Boolean mergeRepeated)
mergeRepeated
- If true, merge repeated classes in output.public List<Output<Long>> decodedIndices()
public List<Output<Long>> decodedValues()
public List<Output<Long>> decodedShape()
Copyright © 2022. All rights reserved.