tf.raw_ops.DynamicEnqueueTPUEmbeddingRaggedTensorBatch

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.DynamicEnqueueTPUEmbeddingRaggedTensorBatch

tf.raw_ops.DynamicEnqueueTPUEmbeddingRaggedTensorBatch(
    sample_splits,
    embedding_indices,
    aggregation_weights,
    mode_override,
    device_ordinal,
    table_ids,
    combiners=[],
    max_sequence_lengths=[],
    num_features=[],
    name=None
)

Аргументы
`sample_splits`	Список как минимум из 1 объектов `Tensor` одного типа: `int32`, `int64`.
`embedding_indices`	Список объектов `Tensor` того же типа, что и в `int32`, `int64`, длина которого совпадает с длиной `sample_splits`.
`aggregation_weights`	Список объектов `Tensor` того же типа, что и в `float32`, `float64`, длина которого совпадает с длиной `sample_splits`.
`mode_override`	`Tensor` типа `string`.
`device_ordinal`	`Tensor` типа `int32`.
`table_ids`	Список из `ints`.
`combiners`	Необязательный список из `strings`. По умолчанию — `[]`.
`max_sequence_lengths`	Необязательный список из `ints`. По умолчанию — `[]`.
`num_features`	Необязательный список из `ints`. По умолчанию — `[]`.
`name`	Имя операции (необязательно).

Возвращаемое значение
Созданный объект Operation.

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