public protocol RNNCell: Layer
where
Input == RNNCellInput<TimeStepInput, State>,
Output == RNNCellOutput<TimeStepOutput, State>Một tế bào mạng lưới thần kinh tái phát.
Đầu vào tại một bước thời gian.
Tuyên ngôn
associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>Đầu ra tại một bước thời gian.
Tuyên ngôn
associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>Trạng thái có thể được bảo toàn qua các bước thời gian.
Tuyên ngôn
associatedtype State : DifferentiableTrả về trạng thái có giá trị bằng 0 với hình dạng tương thích với đầu vào được cung cấp.
Tuyên ngôn
func zeroState(for input: TimeStepInput) -> StateTrả về trạng thái mới thu được từ việc áp dụng ô RNN cho đầu vào ở bước thời gian hiện tại và trạng thái trước đó.
Tuyên ngôn
@differentiable public func callAsFunction( input: TimeStepInput, state: State ) -> RNNCellOutput<TimeStepOutput, State>Thông số
timeStepInputĐầu vào ở bước thời gian hiện tại.
previousStateTrạng thái trước đó của ô RNN.
Giá trị trả về
Đầu ra.
Tuyên ngôn
@differentiable public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>