Spec-Zone.ru › TensorFlow 1.15

tf.contrib.rnn.NASCell

Ячейка рекуррентной сети поиска архитектуры нейронной сети (NAS).

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

tf.contrib.rnn.NASCell(
    num_units, num_proj=None, use_bias=False, reuse=None, **kwargs
)

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

https://arxiv.org/abs/1611.01578

Баррет Зоф и Квок В. Ле. "Поиск архитектуры нейронной сети с помощью обучения с подкреплением" Proc. ICLR 2017.

Класс использует необязательный проекционный слой.

Аргументы
num_units int, количество ячеек в ячейке NAS.
num_proj (необязательно) int, размерность вывода для проекционных матриц. Если None, проекция не выполняется.
use_bias (необязательно) bool, если True, использовать смещения в ячейке. По умолчанию False.
reuse (необязательно) Python-булево значение, описывающее, следует ли повторно использовать переменные в существующем пространстве имен. Если не True, и существующее пространство имен уже содержит заданные переменные, возникает ошибка.
**kwargs Дополнительные ключевые аргументы.
Атрибуты
graph УСТАРЕВШАЯ ФУНКЦИЯ
Предупреждение: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: прекратите использовать это свойство, поскольку слои tf.layers больше не отслеживают свою диаграмму.
output_size Целое число или TensorShape: размер вывода, производимого этой ячейкой.
scope_name
state_size Размер(ы) состояния(ий), используемого(ых) этой ячейкой.

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

Методы

get_initial_state

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

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

zero_state

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

zero_state(
    batch_size, dtype
)

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

Аргументы
batch_size int, float или тензор единиц, представляющий размер пакетной обработки.
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/NASCell

Spec-Zone.ru

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