import { Order } from 'blockly/javascript';
/**
 * @category AI
 * @subcategory Layer
 * @module Recurrent
 */
/**
 * Creates a Long-Short Term Memory layer - Hochreiter 1997.
 *
 * @param {Number} units Positive integer, dimensionality of the output space.
 * @param {Boolean} returnSequence  Whether to return the last output in the output sequence, or the full sequence.
 * @returns {LSTM} tf.layers.lstm({ units, returnSequence }).
 */
function lstmLayer(block, generator) {
    const units = generator.valueToCode(block, 'UNITS', Order.NONE) || '';
    const returnSequence = block.getFieldValue('RETURN') || '';
    return [`tf.layers.lstm({ units: ${units}, returnSequence: ${returnSequence}, })`, Order.VOID];
}
export const layerBlockGenerator = {
    ['add_LSTM_layer']: lstmLayer,
};
//# sourceMappingURL=generators.js.map