RNNCellOutput

public struct RNNCellOutput<Output, State> : Differentiable where Output : Differentiable, State : Differentiable
extension RNNCellOutput: EuclideanDifferentiable
where Output: EuclideanDifferentiable, State: EuclideanDifferentiable

An output to a recurrent neural network.

  • The output at the current time step.

    Declaration

    public var output: Output
  • The current state.

    Declaration

    public var state: State
  • Declaration

    @differentiable
    public init(output: Output, state: State)