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

Возвращаемое значение
`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/api_docs/python/tf/raw_ops/TPUEmbeddingActivations