public struct LSTMCell<Scalar> : RecurrentLayerCell where Scalar : TensorFlowFloatingPoint
Một tế bào LSTM.
Tuyên ngôn
public var fusedWeight: Tensor<Scalar>
Tuyên ngôn
public var fusedBias: Tensor<Scalar>
Tuyên ngôn
public var inputWeight: Tensor<Scalar> { get }
Tuyên ngôn
public var updateWeight: Tensor<Scalar> { get }
Tuyên ngôn
public var forgetWeight: Tensor<Scalar> { get }
Tuyên ngôn
public var outputWeight: Tensor<Scalar> { get }
Tuyên ngôn
public var inputBias: Tensor<Scalar> { get }
Tuyên ngôn
public var updateBias: Tensor<Scalar> { get }
Tuyên ngôn
public var forgetBias: Tensor<Scalar> { get }
Tuyên ngôn
public var outputBias: Tensor<Scalar> { get }
Tuyên ngôn
public typealias TimeStepInput = Tensor<Scalar>
Tuyên ngôn
public typealias TimeStepOutput = State
Tuyên ngôn
public typealias Input = RNNCellInput<TimeStepInput, State>
Tuyên ngôn
public typealias Output = RNNCellOutput<TimeStepOutput, State>
Tạo
LSTMCell
với kích thước đầu vào được chỉ định và kích thước trạng thái ẩn.Tuyên ngôn
public init(inputSize: Int, hiddenSize: Int)
Thông số
inputSize
Số lượng các tính năng trong tensor đầu vào 2-D.
hiddenSize
Số lượng đặc điểm ở trạng thái ẩn 2-D.
Tuyên ngôn
public struct State : Equatable, Differentiable, VectorProtocol, KeyPathIterable, Mergeable