Spec-Zone.ru › TensorFlow 1.15

tf.contrib.cudnn_rnn.CudnnParamsFormatConverterTanh

Класс-помощник, преобразующий параметры Cudnn в параметры TF Tanh RNN.

tf.contrib.cudnn_rnn.CudnnParamsFormatConverterTanh(
    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/CudnnParamsFormatConverterTanh

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API