public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Una cella LSTM.
Dichiarazione
public var fusedWeight: Tensor<Scalar>
Dichiarazione
public var fusedBias: Tensor<Scalar>
Dichiarazione
public var inputWeight: Tensor<Scalar> { get }
Dichiarazione
public var updateWeight: Tensor<Scalar> { get }
Dichiarazione
public var forgetWeight: Tensor<Scalar> { get }
Dichiarazione
public var outputWeight: Tensor<Scalar> { get }
Dichiarazione
public var inputBias: Tensor<Scalar> { get }
Dichiarazione
public var updateBias: Tensor<Scalar> { get }
Dichiarazione
public var forgetBias: Tensor<Scalar> { get }
Dichiarazione
public var outputBias: Tensor<Scalar> { get }
Dichiarazione
public typealias TimeStepInput = Tensor<Scalar>
Dichiarazione
public typealias TimeStepOutput = State
Dichiarazione
public typealias Input = RNNCellInput<TimeStepInput, State>
Dichiarazione
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Crea una
LSTMCell
con la dimensione di input e la dimensione dello stato nascosto specificate.Dichiarazione
public init(inputSize: Int, hiddenSize: Int)
Parametri
inputSize
Il numero di caratteristiche nei tensori di input 2-D.
hiddenSize
Il numero di feature negli stati nascosti 2D.
Dichiarazione
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable