public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Sel LSTM.
Pernyataan
public var fusedWeight: Tensor<Scalar>
Pernyataan
public var fusedBias: Tensor<Scalar>
Pernyataan
public var inputWeight: Tensor<Scalar> { get }
Pernyataan
public var updateWeight: Tensor<Scalar> { get }
Pernyataan
public var forgetWeight: Tensor<Scalar> { get }
Pernyataan
public var outputWeight: Tensor<Scalar> { get }
Pernyataan
public var inputBias: Tensor<Scalar> { get }
Pernyataan
public var updateBias: Tensor<Scalar> { get }
Pernyataan
public var forgetBias: Tensor<Scalar> { get }
Pernyataan
public var outputBias: Tensor<Scalar> { get }
Pernyataan
public typealias TimeStepInput = Tensor<Scalar>
Pernyataan
public typealias TimeStepOutput = State
Pernyataan
public typealias Input = RNNCellInput<TimeStepInput, State>
Pernyataan
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Membuat
LSTMCell
dengan ukuran input yang ditentukan dan ukuran status tersembunyi.Pernyataan
public init(inputSize: Int, hiddenSize: Int)
Parameter
inputSize
Jumlah fitur dalam tensor masukan 2-D.
hiddenSize
Jumlah fitur dalam status tersembunyi 2-D.
Pernyataan
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable