public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Uma célula LSTM.
Declaração
public var fusedWeight: Tensor<Scalar>
Declaração
public var fusedBias: Tensor<Scalar>
Declaração
public var inputWeight: Tensor<Scalar> { get }
Declaração
public var updateWeight: Tensor<Scalar> { get }
Declaração
public var forgetWeight: Tensor<Scalar> { get }
Declaração
public var outputWeight: Tensor<Scalar> { get }
Declaração
public var inputBias: Tensor<Scalar> { get }
Declaração
public var updateBias: Tensor<Scalar> { get }
Declaração
public var forgetBias: Tensor<Scalar> { get }
Declaração
public var outputBias: Tensor<Scalar> { get }
Declaração
public typealias TimeStepInput = Tensor<Scalar>
Declaração
public typealias TimeStepOutput = State
Declaração
public typealias Input = RNNCellInput<TimeStepInput, State>
Declaração
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Cria uma
LSTMCell
com o tamanho de entrada especificado e o tamanho do estado oculto.Declaração
public init(inputSize: Int, hiddenSize: Int)
Parâmetros
inputSize
O número de recursos em tensores de entrada 2D.
hiddenSize
O número de recursos em estados ocultos 2D.
Declaração
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable