Spec-Zone.ru › TensorFlow 1.15

tf.contrib.cudnn_rnn.CudnnGRUSaveable

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

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

Spec-Zone.ru

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