Вычисляет размер весов, которые могут использоваться моделью 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, чтобы сохранить и восстановить их таким образом, чтобы они были совместимы при различных запусках.
Вложенные классы
сорт | CudnnRnnParamsSize.Options | Необязательные атрибуты для CudnnRnnParamsSize |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <U> | какВывод () Возвращает символический дескриптор тензора. |
static <U расширяет TNumber , T расширяет TNumber > CudnnRnnParamsSize <U> | |
статический CudnnRnnParamsSize.Options | направление (направление строки) |
статический CudnnRnnParamsSize.Options | выпадение (выпадение с плавающей запятой) |
статический CudnnRnnParamsSize.Options | inputMode (Строковый режим ввода) |
статический CudnnRnnParamsSize.Options | numProj (длинный numProj) |
Выход <U> | |
статический CudnnRnnParamsSize.Options | rnnMode (строка rnnMode) |
статический CudnnRnnParamsSize.Options | семя (длинное семя) |
статический CudnnRnnParamsSize.Options | семя2 (длинное семя2) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <U> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static CudnnRnnParamsSize <U> create ( Область действия, Операнд <TInt32> numLayers, Операнд <TInt32> numUnits, Операнд <TInt32> inputSize, Class<T> T, Class<U> S, Опции... опции)
Фабричный метод для создания класса, обертывающего новую операцию CudnnRnnParamsSize.
Параметры
объем | текущий объем |
---|---|
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр CudnnRnnParamsSize