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
)

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

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

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: первая часть значения seed для инициализации отбрасывания. seed2: вторая часть значения seed для инициализации отбрасывания. num_proj: Размерность вывода для проекционных матриц. Если None или 0, проекция не выполняется.