public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Une cellule LSTM.
Déclaration
public var fusedWeight: Tensor<Scalar>
Déclaration
public var fusedBias: Tensor<Scalar>
Déclaration
public var inputWeight: Tensor<Scalar> { get }
Déclaration
public var updateWeight: Tensor<Scalar> { get }
Déclaration
public var forgetWeight: Tensor<Scalar> { get }
Déclaration
public var outputWeight: Tensor<Scalar> { get }
Déclaration
public var inputBias: Tensor<Scalar> { get }
Déclaration
public var updateBias: Tensor<Scalar> { get }
Déclaration
public var forgetBias: Tensor<Scalar> { get }
Déclaration
public var outputBias: Tensor<Scalar> { get }
Déclaration
public typealias TimeStepInput = Tensor<Scalar>
Déclaration
public typealias TimeStepOutput = State
Déclaration
public typealias Input = RNNCellInput<TimeStepInput, State>
Déclaration
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Crée un
LSTMCell
avec la taille d'entrée et la taille de l'état caché spécifiées.Déclaration
public init(inputSize: Int, hiddenSize: Int)
Paramètres
inputSize
Le nombre d'entités dans les tenseurs d'entrée 2D.
hiddenSize
Le nombre d'entités dans les états masqués 2D.
Déclaration
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable