tf.raw_ops.TPUEmbeddingActivations

Операция, позволяющая дифференцировать TPU встраивания.

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

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

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

tf.compat.v1.raw_ops.TPUEmbeddingActivations

tf.raw_ops.TPUEmbeddingActivations(
    embedding_variable, sliced_activations, table_id, lookup_id, name=None
)

Эта операция просто возвращает свой первый вход, который предполагается извлеченным из тензоров, возвращаемых TPUEmbeddingDequeueActivations. Наличие этой операции и ее первого аргумента, являющегося обучаемой переменной, позволяет автоматически дифференцировать графики, содержащие встраивания, с помощью библиотек TPU Embedding на Python.

Аргументы
`embedding_variable`	A `Tensor` типа `float32`. Обучаемая переменная, позволяющая оптимизаторам найти эту операцию.
`sliced_activations`	A `Tensor` типа `float32`. Тензор активаций встраивания, который нужно вернуть.
`table_id`	A `int` типа `>= 0`. Идентификатор таблицы в конфигурации слоя встраивания, из которой были вычислены эти активации.
`lookup_id`	A `int` типа `>= 0`. Идентификатор набора индексов встраивания, который вывел эти активации.
`name`	Имя операции (необязательно).

Возвращаемое значение
A `Tensor` типа `float32`.

© 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/raw_ops/TPUEmbeddingActivations