GRU

abstract class GRU : RecurrentBase<GRU, DTensor>

Make a GRU you desire? See invoke in the companion object, or use the GRUEncoder or GRUDecoder helpers.

Types

Companion

object Companion

Functions

accMap

open fun accMap(t: DTensor, sequenceAxis: Int, initialState: DTensor): Pair<DTensor, DTensor>

cell

abstract fun cell(state: Pair<DTensor, DTensor>, x: DTensor): Pair<DTensor, DTensor>

cpu

open override fun cpu(): GRU

doRecurrence

open fun doRecurrence(x: DTensor, initialState: DTensor = this.initialState): DTensor

Do the recurrence.

equals

open operator override fun equals(other: Any?): Boolean

extractTangent

open override fun extractTangent(output: DTensor, extractor: (DTensor, DTensor) -> DTensor): TrainableComponent.Companion.Tangent

fold

open fun fold(t: DTensor, sequenceAxis: Int, initialState: DTensor): Pair<DTensor, DTensor>

getSingleInput

open fun getSingleInput(inputs: Array<out DTensor>): DTensor

Helper to check that the layer was called with a single input. Returns that input if successful, else errors.

gpu

open override fun gpu(): GRU

hashCode

open override fun hashCode(): Int

invoke

open operator override fun invoke(vararg inputs: DTensor): DTensor

load

open override fun load(from: ByteBuffer): GRU

processForBatching

open override fun processForBatching(initialState: DTensor, initialOutput: DTensor, batchSize: Int): Pair<DTensor, DTensor>

store

open override fun store(into: ByteBuffer): ByteBuffer

open fun to(device: Device): OnDevice

trainingStep

open override fun trainingStep(optim: Optimizer<*>, tangent: Trainable.Tangent): GRU

withTrainables

abstract fun withTrainables(trainables: List<Trainable<*>>): GRU

wrap

open override fun wrap(wrapper: Wrapper): GRU

The wrap function should return the same static type it is declared on.

Properties

accType

open override val accType: RecurrentBase.RecurrentBase.AccType

batchAxis

open val batchAxis: Int

initialOutput

open override val initialOutput: FloatTensor

initialState

open override val initialState: DTensor

sequenceAxis

open val sequenceAxis: Int

trainables

abstract val trainables: List<Trainable<*>>

Inheritors

LinearAfterResetGru

LinearBeforeResetGRU