tf.contrib.cudnn_rnn.CudnnLSTM

Реализация слоя LSTM с использованием CUDA.

tf.contrib.cudnn_rnn.CudnnLSTM(
    num_layers, num_units, input_mode=CUDNN_INPUT_LINEAR_MODE,
    direction=CUDNN_RNN_UNIDIRECTION, dropout=0.0, seed=None,
    dtype=tf.dtypes.float32, kernel_initializer=None, bias_initializer=None,
    name=None
)

Аргументы
`num_layers`	количество слоёв в модели RNN.
`num_units`	количество единиц в модели RNN.
`input_mode`	указывает, существует ли линейное преобразование между входом и фактическим вычислением перед первым слоем. Может быть 'linear_input', 'skip_input' или 'auto_select'. 'linear_input' (по умолчанию) всегда применяет линейное преобразование входа к скрытому состоянию RNN. (стандартное поведение RNN). 'skip_input' разрешено только при input_size == num_units; 'auto_select' подразумевает 'skip_input' при input_size == num_units; в противном случае подразумевает 'linear_input'.
`direction`	направление работы модели. Может быть 'unidirectional' или 'bidirectional'
`dropout`	вероятность дропаута, число от [0, 1]. Дропаут применяется между каждым слоем (дропаут не применяется для модели с одним слоем). При значении 0 дропаут отключен.
`seed`	seed для оператора, используемого при инициализации дропаута. См. `tf.compat.v1.set_random_seed` для получения информации о поведении.
`dtype`	tf.float16, tf.float32 или tf.float64
`kernel_initializer`	начальное значение для инициализации весов.
`bias_initializer`	начальное значение для инициализации смещения (по умолчанию все нули).
`name`	VariableScope для созданного подграфа; по умолчанию имя класса. Это служит только для значения по умолчанию, если позже при вызове call() не будет указан scope.

Исключения
`ValueError`	если направление некорректно. Или тип данных не поддерживается.

Атрибуты
`canonical_bias_shapes`	Форматы канонических тензоров смещения Cudnn.
`canonical_weight_shapes`	Форматы канонических тензоров весов Cudnn.
`direction`	Возвращает `unidirectional` или `bidirectional`.
`graph`	УСТАРЕВШАЯ ФУНКЦИЯ Предупреждение: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: перестаньте использовать этот атрибут, т.к. слои tf.layers больше не отслеживают свою графику.
`input_mode`	Режим входных данных первого слоя. Указывает, существует ли линейное преобразование между входом и фактическим вычислением перед первым слоем. Может быть 'linear_input': (по умолчанию) всегда применяет линейное преобразование входа к скрытому состоянию RNN. (стандартное поведение RNN) 'skip_input': 'skip_input' разрешено только при input_size == num_units. 'auto_select': подразумевает 'skip_input' при input_size == num_units; в противном случае подразумевает 'linear_input'.
`input_size`
`num_dirs`
`num_layers`
`num_units`
`rnn_mode`	Тип ячейки RNN, используемой.
`saveable`
`scope_name`

Методы

`state_shape`

Посмотреть исходный код

state_shape(
    batch_size
)

Формат состояний Cudnn LSTM.

Формат - кортеж из 2 элементов. Каждый - [num_layers * num_dirs, batch_size, num_units] Args: batch_size: целое число

Возвращаемое значение
кортеж python-массивов.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/cudnn_rnn/CudnnLSTM