tf.raw_ops.TPUEmbeddingActivations
Операция, позволяющая вычислить производную от TPU встраиваний.
tf.raw_ops.TPUEmbeddingActivations(
embedding_variable, sliced_activations, table_id, lookup_id, name=None
)
Данная операция просто возвращает свой первый входной параметр, который предполагается извлеченным из тензоров, возвращаемых TPUEmbeddingDequeueActivations. Наличие этой операции и ее первого аргумента, являющегося обучаемой переменной, позволяет автоматически вычислять производные в графах, содержащих встраивания, с помощью библиотек TPU Embedding Python.
| Аргументы | |
|---|---|
embedding_variable | Переменная типа float32. Обучаемая переменная, позволяющая оптимизаторам находить эту операцию. |
sliced_activations | Тензор типа float32. Тензор активирования встраиваний, который нужно вернуть. |
table_id | Целочисленное значение >= 0. Идентификатор таблицы в конфигурации слоя встраивания, из которой были вычислены эти активации. |
lookup_id | Целочисленное значение >= 0. Идентификатор набора индексов встраивания, который дал эти активации. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Тензор типа 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
