{ }
Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
tf.raw_ops.DynamicEnqueueTPUEmbeddingArbitraryTensorBatch(
sample_indices_or_row_splits,
embedding_indices,
aggregation_weights,
mode_override,
device_ordinal,
combiners=[],
name=None
)
embedding_indices[i] and aggregation_weights[i] correspond to the 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. |