public protocol RNNCell: Layer
where
Input == RNNCellInput<TimeStepInput, State>,
Output == RNNCellOutput<TimeStepOutput, State>Una célula de red neuronal recurrente.
La entrada en un paso de tiempo.
Declaración
associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>La salida en un paso de tiempo.
Declaración
associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>El estado que puede conservarse a través de pasos de tiempo.
Declaración
associatedtype State : DifferentiableDevuelve un estado de valor cero con una forma compatible con la entrada proporcionada.
Declaración
func zeroState(for input: TimeStepInput) -> StateDevuelve el nuevo estado obtenido al aplicar la celda RNN a la entrada en el paso de tiempo actual y el estado anterior.
Declaración
@differentiable public func callAsFunction( input: TimeStepInput, state: State ) -> RNNCellOutput<TimeStepOutput, State>Parámetros
timeStepInputLa entrada en el paso de tiempo actual.
previousStateEl estado anterior de la celda RNN.
Valor de retorno
La salida.
Declaración
@differentiable public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>