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`	Целое число, являющееся `>= 0`. Идентификатор таблицы в конфигурации слоя встраивания, из которой были вычислены эти активации.
`lookup_id`	Целое число, являющееся `>= 0`. Идентификатор набора индексов встраивания, которые произвели эти активации.
`name`	Имя операции (необязательно).

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

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