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, проекция не выполняется.