tf.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 | Список с длиной, равной sample_splits, содержащий Tensor объекты одного типа в: int32, int64. |
aggregation_weights | Список с длиной, равной sample_splits, содержащий Tensor объекты одного типа в: float32, float64. |
mode_override | Tensor типа string. |
device_ordinal | Tensor типа int32. |
table_ids | Список ints. |
combiners | Необязательный список strings. По умолчанию []. |
max_sequence_lengths | Необязательный список ints. По умолчанию []. |
num_features | Необязательный список ints. По умолчанию []. |
name | Имя операции (необязательно). |
| Возвращаемое значение |
| Созданная операция. |