CudnnRnnParamsSize

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

Возвратите размер параметров, который может использоваться моделью 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; в противном случае подразумевается «линейный_вход». направление: указывает, будет ли использоваться двунаправленная модель. dir = (направление == двунаправленное) ? Выпадение 2:1: вероятность выпадения. Если установлено значение 0, отсев отключен. семя: первая часть семени для инициализации отсева. семя2: вторая часть семени для инициализации отсева. params_size: размер буфера параметров, который должен быть выделен и инициализирован для этой модели RNN. Обратите внимание, что этот буфер параметров может быть несовместим с разными графическими процессорами. Используйте CudnnRNNParamsWeights и CudnnRNNParamsBiases, чтобы сохранить и восстановить их таким образом, чтобы они были совместимы при различных запусках.