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>