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