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