BasicDecoder

Type Members

case class GreedyEmbeddingHelper[S](embeddingFn: (ops.Output) ⇒ ops.Output, beginTokens: ops.Output, endToken: ops.Output, name: String = "RNNDecoderGreedyEmbeddingHelper") extends Helper[ops.Output, S] with Product with Serializable

RNN decoder helper to be used while performing inference.
RNN decoder helper to be used while performing inference. It uses the argmax over the RNN output logits and passes the result through an embedding layer to get the next input.
embeddingFn
Function that takes an INT32 vector of IDs and returns the corresponding embedded values that will be passed to the decoder input.
beginTokens
INT32 vector with length equal to the batch size, which contains the begin-of-sequence token IDs.
endToken
INT32 scalar containing the end-of-sequence token ID (i.e., token ID which marks the end of decoding).
trait Helper[O, S] extends AnyRef

Interface for implementing sampling helpers in sequence-to-sequence decoders.
case class Output[O, OS](rnnOutput: O, sample: O)(implicit whileLoopEvO: Aux[O, OS]) extends Product with Serializable
case class TrainingHelper[O, OS, S, SS](input: O, sequenceLengths: ops.Output, timeMajor: Boolean = false, name: String = "RNNDecoderTrainingHelper")(implicit evO: Aux[O, OS], evS: Aux[S, SS]) extends Helper[O, S] with Product with Serializable

RNN decoder helper to be used while training.
RNN decoder helper to be used while training. It only reads inputs and the returned sample indexes are the argmax over the RNN output logits.

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 Output extends Serializable
def apply[O, OS, S, SS](cell: RNNCell[O, OS, S, SS], initialCellState: S, helper: Helper[O, S], outputLayer: (O) ⇒ O = (o: O) => o, name: String = "BasicRNNDecoder")(implicit evO: Aux[O, OS], evS: Aux[S, SS]): BasicDecoder[O, OS, 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
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 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 BasicDecoder | package decoders

object BasicDecoder

Type Members

case class GreedyEmbeddingHelper[S](embeddingFn: (ops.Output) ⇒ ops.Output, beginTokens: ops.Output, endToken: ops.Output, name: String = "RNNDecoderGreedyEmbeddingHelper") extends Helper[ops.Output, S] with Product with Serializable

trait Helper[O, S] extends AnyRef

case class Output[O, OS](rnnOutput: O, sample: O)(implicit whileLoopEvO: Aux[O, OS]) extends Product with Serializable

case class TrainingHelper[O, OS, S, SS](input: O, sequenceLengths: ops.Output, timeMajor: Boolean = false, name: String = "RNNDecoderTrainingHelper")(implicit evO: Aux[O, OS], evS: Aux[S, SS]) extends Helper[O, S] with Product with Serializable

Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

object Output extends Serializable

def apply[O, OS, S, SS](cell: RNNCell[O, OS, S, SS], initialCellState: S, helper: Helper[O, S], outputLayer: (O) ⇒ O = (o: O) => o, name: String = "BasicRNNDecoder")(implicit evO: Aux[O, OS], evS: Aux[S, SS]): BasicDecoder[O, OS, S, SS]

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

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

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

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