tf.contrib.cudnn_rnn.CudnnParamsFormatConverterRelu
Вспомогательный класс, который преобразует параметры Cudnn и TF Relu RNN.
tf.contrib.cudnn_rnn.CudnnParamsFormatConverterRelu(
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/CudnnParamsFormatConverterRelu