public struct SimpleRNNCell<Scalar> : RNNCell where Scalar : TensorFlowFloatingPoint
A simple RNN cell.
-
Declaration
public var weight: Tensor<Scalar>
-
Declaration
public var bias: Tensor<Scalar>
-
Declaration
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable
-
Declaration
public typealias TimeStepInput = Tensor<Scalar>
-
Declaration
public typealias TimeStepOutput = State
-
Declaration
public typealias Input = RNNCellInput<TimeStepInput, State>
-
Declaration
public typealias Output = RNNCellOutput<TimeStepOutput, State>
-
Creates a
SimpleRNNCell
with the specified input size and hidden state size.Declaration
public init(inputSize: Int, hiddenSize: Int, seed: TensorFlowSeed = Context.local.randomSeed)
Parameters
inputSize
The number of features in 2-D input tensors.
hiddenSize
The number of features in 2-D hidden states.
seed
The random seed for initialization. The default value is random.