public protocol RNNCell: Layer
where
Input == RNNCellInput<TimeStepInput, State>,
Output == RNNCellOutput<TimeStepOutput, State>Powtarzająca się komórka sieci neuronowej.
Wejście w kroku czasowym.
Deklaracja
associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>Dane wyjściowe w kroku czasowym.
Deklaracja
associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>Stan, który może być zachowany w różnych etapach czasowych.
Deklaracja
associatedtype State : DifferentiableZwraca stan o wartości zerowej o kształcie zgodnym z podanymi danymi wejściowymi.
Deklaracja
func zeroState(for input: TimeStepInput) -> StateZwraca nowy stan uzyskany po zastosowaniu komórki RNN do wejścia w bieżącym kroku czasowym i w poprzednim stanie.
Deklaracja
@differentiable public func callAsFunction( input: TimeStepInput, state: State ) -> RNNCellOutput<TimeStepOutput, State>Parametry
timeStepInputWejście w bieżącym kroku czasowym.
previousStatePoprzedni stan komórki RNN.
Wartość zwracana
Wyjście.
Deklaracja
@differentiable public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>