Spec-Zone.ru › TensorFlow 1.15

tf.contrib.cudnn_rnn.CudnnRNNReluSaveable

Реализация SaveableObject для обработки неявных параметров Cudnn LSTM.

Наследуется от: CudnnLSTMSaveable

tf.contrib.cudnn_rnn.CudnnRNNReluSaveable(
    opaque_params, num_layers, num_units, input_size,
    input_mode=CUDNN_INPUT_LINEAR_MODE, direction=CUDNN_RNN_UNIDIRECTION,
    scope=None, name='cudnn_rnn_saveable'
)
Аргументы
opaque_params переменная, неявные параметры Cudnn RNN.
num_layers количество слоёв для модели RNN.
num_units количество единиц в модели RNN.
input_size размер входных данных, он может отличаться от num_units.
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'
scope строка VariableScope, область эквивалентной подграфа, состоящая только из независимых от платформы ячеек tf RNN.
name имя объекта CudnnOpaqueParamsSaveable.
Атрибуты
device Устройство для тензоров SaveSpec.
format_converter
optional_restore Подсказка для восстановления утверждений, что этот объект является необязательным.

Методы

restore

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

restore(
    restored_tensors, restored_shapes
)

Восстанавливает этот объект из 'restored_tensors'.

Аргументы
restored_tensors тензоры, загруженные из контрольной точки
restored_shapes формы, которым этот объект должен соответствовать после восстановления, или None.
Возвращает
Операция, восстанавливающая состояние объекта.
Исключения
ValueError Если объект не может быть восстановлен с помощью предоставленных параметров.

© 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/CudnnRNNReluSaveable

Spec-Zone.ru

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