public protocol RecurrentLayerCell: Layer
where
Input == RNNCellInput<TimeStepInput, State>,
Output == RNNCellOutput<TimeStepOutput, State>Une cellule de couche récurrente.
L'entrée à un pas de temps.
Déclaration
associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>La sortie à un pas de temps.
Déclaration
associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>État qui peut être préservé au fil du temps.
Déclaration
associatedtype State : DifferentiableRenvoie un état de valeur zéro avec une forme compatible avec l'entrée fournie.
Déclaration
func zeroState(for input: TimeStepInput) -> StateRenvoie le nouvel état obtenu en appliquant la cellule de couche récurrente à l'entrée au pas de temps actuel et à l'état précédent.
Déclaration
@differentiable
public func callAsFunction(
input: TimeStepInput,
state: State
) -> RNNCellOutput<TimeStepOutput, State>Paramètres
timeStep Input L’entrée au pas de temps actuel.
previousStateL'état précédent de la cellule de la couche récurrente.
Valeur de retour
La sortie.
Déclaration
@differentiable
public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>