public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Ячейка LSTM.
Декларация
public var fusedWeight: Tensor<Scalar>
Декларация
public var fusedBias: Tensor<Scalar>
Декларация
public var inputWeight: Tensor<Scalar> { get }
Декларация
public var updateWeight: Tensor<Scalar> { get }
Декларация
public var forgetWeight: Tensor<Scalar> { get }
Декларация
public var outputWeight: Tensor<Scalar> { get }
Декларация
public var inputBias: Tensor<Scalar> { get }
Декларация
public var updateBias: Tensor<Scalar> { get }
Декларация
public var forgetBias: Tensor<Scalar> { get }
Декларация
public var outputBias: Tensor<Scalar> { get }
Декларация
public typealias TimeStepInput = Tensor<Scalar>
Декларация
public typealias TimeStepOutput = State
Декларация
public typealias Input = RNNCellInput<TimeStepInput, State>
Декларация
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Создает
LSTMCell
с указанным входным размером и размером скрытого состояния.Декларация
public init(inputSize: Int, hiddenSize: Int)
Параметры
inputSize
Количество функций в двумерных входных тензорах.
hiddenSize
Количество функций в скрытых двумерных состояниях.
Декларация
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable