tf.raw_ops.CudnnRNNParamsToCanonicalV2

Возвращает параметры CudnnRNN в канонической форме. Поддерживает проекцию в LSTM.

tf.raw_ops.CudnnRNNParamsToCanonicalV2(
    num_layers,
    num_units,
    input_size,
    params,
    num_params_weights,
    num_params_biases,
    rnn_mode='lstm',
    input_mode='linear_input',
    direction='unidirectional',
    dropout=0,
    seed=0,
    seed2=0,
    num_proj=0,
    name=None
)

Возвращает набор весов из неявного буфера параметров, которые можно сохранять и восстанавливать в формате, совместимом с будущими запусками.

Обратите внимание, что буфер параметров может быть несовместим с разными графическими процессорами. Поэтому любое сохранение и восстановление должно быть преобразовано в канонические веса и смещения.

num_layers: Указывает количество слоев в модели RNN. num_units: Указывает размер скрытого состояния. input_size: Указывает размер входного состояния. num_params_weights: количество матриц весовых параметров для всех слоёв. num_params_biases: количество векторов смещений параметров для всех слоёв. weights: каноническая форма весов, которая может использоваться для сохранения и восстановления. Они, скорее всего, будут совместимы с разными поколениями. biases: каноническая форма смещений, которая может использоваться для сохранения и восстановления. Они, скорее всего, будут совместимы с разными поколениями. rnn_mode: Указывает тип модели RNN. input_mode: Указывает, есть ли линейная проекция между входом и фактическим вычислением перед первым слоем. `skip_input` разрешено только тогда, когда `input_size == num_units`; `auto_select` подразумевает `skip_input`, когда `input_size == num_units`; в противном случае он подразумевает `linear_input`. direction: Указывает, будет ли использоваться двунаправленная модель. dir = (direction == bidirectional) ? 2 : 1 dropout: вероятность дропаута. При значении 0. дропаут отключен. seed: первая часть семени для инициализации дропаута. seed2: вторая часть семени для инициализации дропаута. num_proj: размерность вывода для проекционных матриц. Если None или 0, проекция не выполняется.