BeamSearchDecoder

Type Members

case class FinalOutput(predictedIDs: ops.Output, output: Output) extends Product with Serializable

Final outputs returned by the beam search after all decoding is finished.
Final outputs returned by the beam search after all decoding is finished.
predictedIDs
Tensor of shape [batchSize, T, beamWidth] (or [T, batchSize, beamWidth], if outputTimeMajor == true) containing the final prediction IDs. The search states are ordered from best to worst.
output
State of the beam search at the end of decoding.
case class Output(scores: ops.Output, predictedIDs: ops.Output, parentIDs: ops.Output) extends Product with Serializable
case class State[S, SS](rnnState: S, logProbabilities: ops.Output, sequenceLengths: ops.Output, finished: ops.Output)(implicit evS: Aux[S, SS]) extends Product with Serializable

Value Members

final def !=(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def ##(): Int

Definition Classes
AnyRef → Any
final def ==(arg0: Any): Boolean

Definition Classes
AnyRef → Any
object FinalOutput extends Serializable
object Output extends Serializable
object State extends Serializable
def apply[S, SS](cell: RNNCell[ops.Output, core.Shape, S, SS], initialCellState: S, embeddingFn: (ops.Output) ⇒ ops.Output, beginTokens: ops.Output, endToken: ops.Output, beamWidth: Int, lengthPenalty: LengthPenalty = NoPenalty, outputLayer: (ops.Output) ⇒ ops.Output = (o: ops.Output) => o, reorderTensorArrays: Boolean = true, name: String = "BeamSearchRNNDecoder")(implicit evOutput: Aux[ops.Output, core.Shape], evS: Aux[S, SS]): BeamSearchDecoder[S, SS]
final def asInstanceOf[T0]: T0

Definition Classes
Any
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
final def eq(arg0: AnyRef): Boolean

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

Definition Classes
AnyRef → Any
def finalize(): Unit

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( classOf[java.lang.Throwable] )
final def getClass(): Class[_]

Definition Classes
AnyRef → Any
def hashCode(): Int

Definition Classes
AnyRef → Any
final def isInstanceOf[T0]: Boolean

Definition Classes
Any
val logger: Logger

Attributes
protected
def maybeSplitBatchBeams(value: OutputConvertible, shape: core.Shape, batchSize: ops.Output, beamWidth: Int): OutputConvertible

Maybe converts the provided tensor structure from batches by beams into batches of beams, by merging them accordingly.
Maybe converts the provided tensor structure from batches by beams into batches of beams, by merging them accordingly.
More precisely, value consists of tensors with shape [batchSize * beamWidth] ++ ... and this method reshapes them into tensors with shape [batchSize, beamWidth] ++ ....
value
Value to reshape.
shape
Depth shape of the value.
returns
Reshaped state.

Annotations
@throws( ... )
Exceptions thrown
InvalidArgumentException If value is of an unsupported type, or if it contains any tensors of unknown rank, or if, after reshaping, the new tensor is not shaped [batchSize, beamWidth] ++ ... (assuming that both batchSize and beamWidth are known statically).
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def tileForBeamSearch[S](value: S, beamWidth: Int)(implicit evS: WhileLoopVariable[S]): S

Annotations
@throws( ... )
def toString(): String

Definition Classes
AnyRef → Any
final def wait(): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long, arg1: Int): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )

Related Docs: class BeamSearchDecoder | package decoders

object BeamSearchDecoder

Type Members

case class FinalOutput(predictedIDs: ops.Output, output: Output) extends Product with Serializable

case class Output(scores: ops.Output, predictedIDs: ops.Output, parentIDs: ops.Output) extends Product with Serializable

case class State[S, SS](rnnState: S, logProbabilities: ops.Output, sequenceLengths: ops.Output, finished: ops.Output)(implicit evS: Aux[S, SS]) extends Product with Serializable

Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

object FinalOutput extends Serializable

object Output extends Serializable

object State extends Serializable

final def asInstanceOf[T0]: T0

def clone(): AnyRef

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def finalize(): Unit

final def getClass(): Class[_]

def hashCode(): Int

final def isInstanceOf[T0]: Boolean

val logger: Logger

def maybeSplitBatchBeams(value: OutputConvertible, shape: core.Shape, batchSize: ops.Output, beamWidth: Int): OutputConvertible

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

final def synchronized[T0](arg0: ⇒ T0): T0

def tileForBeamSearch[S](value: S, beamWidth: Int)(implicit evS: WhileLoopVariable[S]): S

def toString(): String

final def wait(): Unit

final def wait(arg0: Long, arg1: Int): Unit

final def wait(arg0: Long): Unit

Inherited from AnyRef

Inherited from Any

Ungrouped