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