CudnnRNN

публичный финальный класс CudnnRNN

RNN, поддерживаемый cuDNN.

Вычисляет RNN на основе входного и начального состояний относительно буфера параметров. Принимает один дополнительный ввод «sequence_lengths», чем CudnnRNN.

rnn_mode: указывает тип модели RNN. input_mode: указывает, существует ли линейная проекция между входными данными и фактическими вычислениями перед первым слоем. «skip_input» разрешен только в том случае, если input_size == num_units; «auto_select» подразумевает «skip_input», когда input_size == num_units; в противном случае подразумевается «линейный_вход». направление: указывает, будет ли использоваться двунаправленная модель. Должен быть «однонаправленным» или «двунаправленным». отсев: вероятность отсева. Если установлено значение 0, отсев отключен. семя: первая часть семени для инициализации отсева. семя2: вторая часть начального числа для инициализации отсева. ввод: если time_major имеет значение true, это трехмерный тензор формы [seq_length, Batch_size, input_size]. Если time_major имеет значение false, форма имеет вид [batch_size, seq_length, input_size]. input_h: Если time_major имеет значение true, это трехмерный тензор формы [num_layer * dir, package_size, num_units]. Если time_major имеет значение false, форма имеет вид [batch_size, num_layer * dir, num_units]. input_c: для LSTM — трехмерный тензор формы [num_layer * dir, пакет, num_units]. Для других моделей он игнорируется. params: одномерный тензор, который содержит веса и смещения в непрозрачном макете. Размер должен быть создан с помощью CudnnRNNParamsSize и инициализирован отдельно. Обратите внимание, что они могут быть несовместимы в разных поколениях. Поэтому рекомендуется сохранить и восстановить последовательность_длин: вектор длин каждой входной последовательности. вывод: если time_major имеет значение true, это трехмерный тензор формы [seq_length, Batch_size, dir * num_units]. Если time_major имеет значение false, форма имеет вид [batch_size, seq_length, dir * num_units]. output_h: та же форма имеет input_h. output_c: та же форма, что и input_c для LSTM. Пустой тензор для других моделей. is_training: указывает, используется ли эта операция для вывода или обучения. time_major: указывает, является ли формат ввода/вывода основным или пакетным. резерв_пространство: непрозрачный тензор, который можно использовать при расчете обратного распространения. Он создается только в том случае, если is_training имеет значение true.

Вложенные классы

сорт CudnnRNN.Параметры Дополнительные атрибуты для CudnnRNN

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

Публичные методы

static <T расширяет TNumber > CudnnRNN <T>
create ( Область области действия, ввод операнда <T>, ввод операнда <T> inputH, операнд <T> inputC, параметры операнда <T>, параметры операнда < TInt32 > SequenceLengths, параметры... )
Фабричный метод для создания класса, обертывающего новую операцию CudnnRNN.
статический CudnnRNN.Options
направление (направление строки)
статический CudnnRNN.Options
выпадение (выпадение с плавающей запятой)
Вывод <?>
статический CudnnRNN.Options
inputMode (Строковый режим ввода)
статический CudnnRNN.Options
isTraining (логическое значение isTraining)
статический CudnnRNN.Options
numProj (длинный numProj)
Выход <Т>
Выход <Т>
Выход <Т>
Выход <Т>
статический CudnnRNN.Options
rnnMode (строка rnnMode)
статический CudnnRNN.Options
семя (длинное семя)
статический CudnnRNN.Options
семя2 (длинное семя2)
статический CudnnRNN.Options
timeMajor (логическое значение timeMajor)

Унаследованные методы

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «CudnnRNNV3».

Публичные методы

public static CudnnRNN <T> create ( Область действия, ввод операнда <T>, операнд <T> inputH, операнд <T> inputC, параметры операнда <T>, операнд < TInt32 > последовательностьLengths, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию CudnnRNN.

Параметры
объем текущий объем
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр CudnnRNN

общедоступное статическое направление CudnnRNN.Options (направление строки)

общедоступный статический отказ CudnnRNN.Options (выпадение с плавающей запятой)

публичный вывод <?> hostReserved ()

общедоступный статический CudnnRNN.Options inputMode (String inputMode)

общедоступный статический CudnnRNN.Options isTraining (Boolean isTraining)

общедоступный статический CudnnRNN.Options numProj (длинный numProj)

публичный вывод <T> вывод ()

публичный вывод <T> outputC ()

публичный вывод <T> outputH ()

публичный вывод <T> ReserveSpace ()

общедоступный статический CudnnRNN.Options rnnMode (String rnnMode)

публичное статическое семя CudnnRNN.Options (длинное семя)

общедоступный статический CudnnRNN.Options семя2 (длинное семя2)

общедоступный статический CudnnRNN.Options timeMajor (логическое значение timeMajor)