public protocol RecurrentLayerCell: Layer
where
Input == RNNCellInput<TimeStepInput, State>,
Output == RNNCellOutput<TimeStepOutput, State>
再帰層セル。
タイムステップでの入力。
宣言
associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>
タイムステップでの出力。
宣言
associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>
タイムステップ全体で保存できる状態。
宣言
associatedtype State : Differentiable
指定された入力と互換性のある形状を持つゼロ値の状態を返します。
宣言
func zeroState(for input: TimeStepInput) -> State
現在のタイム ステップでの入力にリカレント レイヤー セルを適用することによって取得された新しい状態と前の状態を返します。
宣言
@differentiable public func callAsFunction( input: TimeStepInput, state: State ) -> RNNCellOutput<TimeStepOutput, State>
パラメータ
timeStepInput
現在のタイムステップでの入力。
previousState
リカレント層セルの前の状態。
戻り値
出力。
宣言
@differentiable public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>