Contents

init(descriptor:inputWeights:hiddenWeights:peepholeWeights:biases:)

Creates an LSTM layer with the descriptor, weights, and biases you specify.

Declaration

convenience init?(descriptor: MLCLSTMDescriptor, inputWeights: [MLCTensor], hiddenWeights: [MLCTensor], peepholeWeights: [MLCTensor]?, biases: [MLCTensor]?)

Parameters

  • descriptor:

    An object you use to configure the LSTM layer.

  • inputWeights:

    An array that contains tensors that describe the input weights.

  • hiddenWeights:

    An array that contains tensors that describe the hidden weights.

  • peepholeWeights:

    An array that contains tensors that describe the peephole weights.

  • biases:

    An array that contains tensors that describe the bias terms.

See Also

Creating LSTM Layers