Spec-Zone.ru › TensorFlow 1.15

tf.contrib.cudnn_rnn.CudnnRNNTanhSaveable

Реализация SaveableObject, обрабатывающая неявные параметры Cudnn LSTM.

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

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

Spec-Zone.ru

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