Spec-Zone.ru › TensorFlow 2.3

tf.raw_ops.CudnnRNNCanonicalToParams

Преобразует параметры CudnnRNN из канонической формы в пригодную для использования.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.CudnnRNNCanonicalToParams

tf.raw_ops.CudnnRNNCanonicalToParams(
    num_layers, num_units, input_size, weights, biases, rnn_mode='lstm',
    input_mode='linear_input', direction='unidirectional', dropout=0, seed=0,
    seed2=0, name=None
)

Записывает набор весов в буфер параметров, чтобы их можно было использовать в последующих этапах обучения или вывода.

Обратите внимание, что буфер параметров может не быть совместимым с разными графическими процессорами. Поэтому любое сохранение и восстановление должно преобразовываться в канонические веса и смещения.

num_layers: Указывает количество слоёв в модели RNN. num_units: Указывает размер скрытого состояния. input_size: Указывает размер входного состояния. weights: каноническая форма весов, которая может быть использована для сохранения и восстановления. Они, скорее всего, будут совместимы с различными поколениями. biases: каноническая форма смещений, которая может быть использована для сохранения и восстановления. Они, скорее всего, будут совместимы с различными поколениями. num_params: количество наборов параметров для всех слоёв. Каждый слой может содержать несколько наборов параметров, каждый из которых состоит из матрицы весов и вектора смещений. rnn_mode: Указывает тип модели RNN. input_mode: Указывает, существует ли линейная проекция между входом и фактическим вычислением перед первым слоем. «skip_input» разрешен только при input_size == num_units; «auto_select» подразумевает «skip_input», когда input_size == num_units; в противном случае подразумевает «linear_input». direction: Указывает, будет ли использоваться двунаправленная модель. dir = (direction == bidirectional) ? 2 : 1 dropout: вероятность дропаута. При значении 0. дропаут отключен. seed: первая часть генератора случайных чисел для инициализации дропаута. seed2: вторая часть генератора случайных чисел для инициализации дропаута.

Аргументы
num_layers A Tensor типа int32.
num_units A Tensor типа int32.
input_size A Tensor типа int32.
weights Список из как минимум 1 объекта Tensor одинакового типа в: half, float32, float64.
biases Список с тем же количеством элементов, что и у weights, объектов Tensor одного типа с типом weights.
rnn_mode Необязательное значение string из: "rnn_relu", "rnn_tanh", "lstm", "gru". По умолчанию "lstm".
input_mode Необязательное значение string из: "linear_input", "skip_input", "auto_select". По умолчанию "linear_input".
direction Необязательное значение string из: "unidirectional", "bidirectional". По умолчанию "unidirectional".
dropout Необязательное float. По умолчанию 0.
seed Необязательное int. По умолчанию 0.
seed2 Необязательное int. По умолчанию 0.
name Название операции (необязательно).
Возвращаемое значение
A Tensor. Имеет тот же тип, что и weights.

© 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/r2.3/api_docs/python/tf/raw_ops/CudnnRNNCanonicalToParams

Spec-Zone.ru

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