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
제공된 입력과 호환되는 모양의 값이 0인 상태를 반환합니다.
선언
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>