tf.raw_ops.CudnnRNNParamsSize

Вычисляет размер весов, которые могут использоваться моделью Cudnn RNN.

tf.raw_ops.CudnnRNNParamsSize(
    num_layers, num_units, input_size, T, S, rnn_mode='lstm',
    input_mode='linear_input', direction='unidirectional',
    dropout=0, seed=0, seed2=0, num_proj=0, name=None
)

Возвращает размер параметров, который может использоваться моделью Cudnn RNN. Последующее выделение и инициализация весов должны использовать этот размер.

num_layers: Указывает количество слоёв в модели RNN. num_units: Указывает размер скрытого состояния. input_size: Указывает размер входного состояния. 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 для инициализации дропаута. params_size: Размер буфера параметров, который должен быть выделен и инициализирован для этой модели RNN. Обратите внимание, что этот буфер параметров может быть несовместим на разных графических процессорах. Используйте CudnnRNNParamsWeights и CudnnRNNParamsBiases для сохранения и восстановления данных в формате, совместимом с различными запусками.