tf.compat.v1.nn.rnn_cell.DeviceWrapper

Оператор, обеспечивающий выполнение RNNCell на конкретном устройстве.

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

tf.compat.v1.nn.rnn_cell.DeviceWrapper(
    cell, device, **kwargs
)

Аргументы
`cell`	Экземпляр `RNNCell`.
`device`	Строка устройства или функция для передачи в `tf.device`.
`**kwargs`	Словарь аргументов ключевых слов для базового слоя.

Атрибуты
`graph`
`output_size`	Целое число или TensorShape: размер выходных данных, создаваемых этим ячейкой.
`scope_name`
`state_size`	Размер(ы) состояния(й), используемого(ых) этой ячейкой. Может быть представлен целым числом, TensorShape или кортежем целых чисел или TensorShapes.

Методы

`apply`

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

apply(
    *args, **kwargs
)

`get_initial_state`

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

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

`get_losses_for`

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

get_losses_for(
    inputs
)

Извлекает потери, относящиеся к конкретному набору входных данных.

Аргументы
`inputs`	Входной тензор или список/кортеж входных тензоров.

Возвращает
Список тензоров потерь слоя, которые зависят от `inputs`.

`get_updates_for`

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

get_updates_for(
    inputs
)

Извлекает обновления, относящиеся к конкретному набору входных данных.

Аргументы
`inputs`	Входной тензор или список/кортеж входных тензоров.

Возвращает
Список операций обновления слоя, которые зависят от `inputs`.

`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`.

Если state_size является целым числом или TensorShape, то возвращаемое значение представляет собой тензор N-D формы [batch_size, state_size], заполненный нулями.

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

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/nn/rnn_cell/DeviceWrapper