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 | УСТАНОВЛЕНА ФУНКЦИЯ |
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], заполненный нулями. Если |
© 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