Spec-Zone.ru › TensorFlow 1.15

tf.contrib.rnn.UGRNNCell

Ячейка рекуррентной нейронной сети с обновляющим шлюзом (UGRNN).

Наследуется от: RNNCell

tf.contrib.rnn.UGRNNCell(
    num_units, initializer=None, forget_bias=1.0, activation=tf.math.tanh,
    reuse=None
)

Компромисс между LSTM/GRU и обычной RNN. Есть только один шлюз, который определяет, должна ли единица интегрировать или вычислять мгновенно. Это рекуррентная идея сети «Highway Network».

Реализует рекуррентную ячейку из статьи:

https://arxiv.org/abs/1611.09913

Jasmine Collins, Jascha Sohl-Dickstein и David Sussillo. «Емкость и обучаемость в рекуррентных нейронных сетях» Proc. ICLR 2017.

Аргументы
num_units int, Количество единиц в ячейке UGRNN
initializer (необязательно) Инициализатор для матриц весов.
forget_bias (необязательно) float, по умолчанию 1.0, Начальный смещение шлюза забывания, используется для уменьшения масштаба забывания в начале обучения.
activation (необязательно) Функция активации внутренних состояний. По умолчанию tf.tanh.
reuse (необязательно) Логическое значение, описывающее, нужно ли повторно использовать переменные в существующем контексте. Если нет True, и в существующем контексте уже есть заданные переменные, возникает ошибка.
Атрибуты
graph УСТАНОВЛЕНА ФУНКЦИЯ
Предупреждение: ЭТА ФУНКЦИЯ УСТАНОВЛЕНА. Она будет удалена в будущей версии. Инструкции по обновлению: прекратите использование этого свойства, так как слои tf.layers больше не отслеживают свой график.
output_size Целое число или TensorShape: размер выходных данных, производимых этой ячейкой.
scope_name
state_size размер(ы) состояния(ий), используемых этой ячейкой.

Его можно представить целым числом, TensorShape или кортежем целых чисел или TensorShapes.

Методы

get_initial_state

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

get_initial_state(
    inputs=None, batch_size=None, dtype=None
)

zero_state

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

zero_state(
    batch_size, dtype
)

Возвращает тензор(ы) состояния, заполненные нулями.

Аргументы
batch_size целое число, число с плавающей точкой или тензор единиц, представляющий размер пакета.
dtype тип данных для состояния.
Возвращаемое значение
Если state_size является целым числом или TensorShape, то возвращаемое значение — это тензор N-D формы [batch_size, state_size], заполненный нулями.

Если state_size — вложенный список или кортеж, то возвращаемое значение — вложенный список или кортеж (с той же структурой) тензоров 2-D с формами [batch_size, s] для каждого s в state_size.

© 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/rnn/UGRNNCell

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API