{ }
Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
tf.raw_ops.EnqueueTPUEmbeddingSparseTensorBatch(
sample_indices,
embedding_indices,
aggregation_weights,
mode_override,
table_ids,
device_ordinal=-1,
combiners=[],
max_sequence_lengths=[],
num_features=[],
name=None
)
sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond to the ith feature. table_ids[i] indicates which embedding table to look up ith feature.
The tensors at corresponding positions in the three input lists (sample_indices, embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1 with dim_size() equal to the total number of lookups into the table described by the corresponding feature.
Returns | |
---|---|
The created Operation. |