tf.contrib.cudnn_rnn.CudnnParamsFormatConverterLSTM

Вспомогательный класс, преобразующий параметры Cudnn в параметры TF LSTM.

tf.contrib.cudnn_rnn.CudnnParamsFormatConverterLSTM(
    num_layers, num_units, input_size, num_proj=None,
    input_mode=CUDNN_INPUT_LINEAR_MODE, direction=CUDNN_RNN_UNIDIRECTION
)

Аргументы
`num_layers`	количество слоёв для модели RNN.
`num_units`	количество единиц в модели RNN.
`input_size`	размер входных данных, может отличаться от num_units.
`num_proj`	размерность вывода для проекционных матриц. Если None или 0, проекция не выполняется.
`input_mode`	указывает, есть ли линейная проекция между входом и фактическим вычислением перед первым слоем. Может быть 'linear_input', 'skip_input' или 'auto_select'. * 'linear_input' (по умолчанию) всегда применяет линейную проекцию входа на скрытое состояние RNN. (стандартное поведение RNN). * 'skip_input' разрешается только при input_size == num_units; * 'auto_select' подразумевает 'skip_input' при input_size == num_units; в противном случае подразумевает 'linear_input'.
`direction`	направление модели, в котором работает модель. Может быть 'unidirectional' или 'bidirectional'.

Методы

`opaque_to_tf_canonical`

Просмотреть исходный код

opaque_to_tf_canonical(
    opaque_param
)

Преобразует неявные параметры cudnn в стандартные веса tf.

`tf_canonical_to_opaque`

Просмотреть исходный код

tf_canonical_to_opaque(
    tf_canonicals, weights_proj=None
)

Преобразует стандартные веса tf в неявные параметры cudnn.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/cudnn_rnn/CudnnParamsFormatConverterLSTM