RNN

public struct RNN<Cell> : Layer where Cell : RNNCell
extension RNN: Equatable where Cell: Equatable
extension RNN: AdditiveArithmetic where Cell: AdditiveArithmetic
  • Declaration

    public typealias Input = [Cell.TimeStepInput]
  • Declaration

    public typealias Output = [Cell.TimeStepOutput]
  • Declaration

    public var cell: Cell
  • Declaration

    public init(_ cell: @autoclosure () -> Cell)
  • Declaration

    @differentiable
    public func callAsFunction(
      _ inputs: [Cell.TimeStepInput],
      initialState: Cell.State
    ) -> [Cell.TimeStepOutput]
  • Declaration

    @differentiable
    public func call(
      _ inputs: [Cell.TimeStepInput],
      initialState: Cell.State
    ) -> [Cell.TimeStepOutput]
  • Declaration

    @differentiable
    public func callAsFunction(_ inputs: [Cell.TimeStepInput]) -> [Cell.TimeStepOutput]
  • Declaration

    @differentiable
    public func lastOutput(
      from inputs: [Cell.TimeStepInput],
      initialState: Cell.State
    ) -> Cell.TimeStepOutput
  • Declaration

    @differentiable
    public func lastOutput(from inputs: [Cell.TimeStepInput]) -> Cell.TimeStepOutput