init(inputSize:hiddenSize:layerCount:)
Creates a batch first LSTM descriptor with the input size and number of layers you specify.
Declaration
convenience init(inputSize: Int, hiddenSize: Int, layerCount: Int)Parameters
- inputSize:
The number of expected features in the input.
- hiddenSize:
The number of features in the hidden state.
- layerCount:
The number of recurrent layers.
See Also
Creating LSTM Descriptors
init(inputSize:hiddenSize:layerCount:usesBiases:isBidirectional:dropout:)init(inputSize:hiddenSize:layerCount:usesBiases:batchFirst:isBidirectional:dropout:)init(inputSize:hiddenSize:layerCount:usesBiases:batchFirst:isBidirectional:returnsSequences:dropout:)init(inputSize:hiddenSize:layerCount:usesBiases:batchFirst:isBidirectional:returnsSequences:dropout:resultMode:)