GRUCell

public var updateKernel: Tensor<Scalar>

public var updateRecurrentKernel: Tensor<Scalar>

public var resetKernel: Tensor<Scalar>

public var resetRecurrentKernel: Tensor<Scalar>

public var outputKernel: Tensor<Scalar>

public var outputRecurrentKernel: Tensor<Scalar>

public var updateBias: Tensor<Scalar>

public var updateRecurrentBias: Tensor<Scalar>

public var resetBias: Tensor<Scalar>

public var resetRecurrentBias: Tensor<Scalar>

public var outputBias: Tensor<Scalar>

public var outputRecurrentBias: Tensor<Scalar>

@noDerivative
public var stateShape: TensorShape { get }

public func zeroState(for input: Tensor<Scalar>) -> State

public typealias State = Tensor<Scalar>

public typealias TimeStepInput = Tensor<Scalar>

public typealias TimeStepOutput = State

public typealias Input = RNNCellInput<TimeStepInput, State>

public typealias Output = RNNCellOutput<TimeStepOutput, State>

public init(
  inputSize: Int,
  hiddenSize: Int,
  kernelInitializer: ParameterInitializer<Scalar> = glorotUniform(),
  biasInitializer: ParameterInitializer<Scalar> = zeros()
)

@differentiable
public func callAsFunction(_ input: Input) -> Output