public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPointAn LSTM cell.
-
Declaration
public var fusedWeight: Tensor<Scalar> -
Declaration
public var fusedBias: Tensor<Scalar> -
Declaration
public var inputWeight: Tensor<Scalar> { get } -
Declaration
public var updateWeight: Tensor<Scalar> { get } -
Declaration
public var forgetWeight: Tensor<Scalar> { get } -
Declaration
public var outputWeight: Tensor<Scalar> { get } -
Declaration
public var inputBias: Tensor<Scalar> { get } -
Declaration
public var updateBias: Tensor<Scalar> { get } -
Declaration
public var forgetBias: Tensor<Scalar> { get } -
Declaration
public var outputBias: Tensor<Scalar> { get } -
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
LSTMCellwith the specified input size and hidden state size.Declaration
public init(inputSize: Int, hiddenSize: Int)Parameters
inputSizeThe number of features in 2-D input tensors.
hiddenSizeThe number of features in 2-D hidden states.
-
Declaration
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable