MLCLSTMLayer
A layer that represents long short-term memory (LSTM) networks.
Declaration
class MLCLSTMLayerOverview
Use this class to create an LSTM layer with one of the following configurations:
- Unidirectional single layer
The input weights, hidden weights, and biases are arrays of 4 tensors that describe the specified weights for the input, hidden, cell, and output gates.
- Unidirectional stacked layers
The input weights, hidden weights and biases are arrays of layerCount
* 4tensors that describe the specified weights for the input, hidden, cell, and output gates, forlayer0...layer(layerCount - 1).- Bidirectional single layer
The input weights, hidden weights, and biases are arrays of 8 tensors, where the backward time weights and biases follow the forward time weights.
Topics
Creating LSTM Layers
init(descriptor:inputWeights:hiddenWeights:biases:)init(descriptor:inputWeights:hiddenWeights:peepholeWeights:biases:)init(descriptor:inputWeights:hiddenWeights:peepholeWeights:biases:gateActivations:outputResultActivation:)MLCLSTMDescriptorMLCLSTMResultMode