(embeddings, segment_ids, column_ids, sp_shape)
| 470 | |
| 471 | |
| 472 | def _tile_combine_embedding(embeddings, segment_ids, column_ids, sp_shape): |
| 473 | column_ids = math_ops.cast(column_ids, dtypes.int32) |
| 474 | sp_shape = math_ops.cast(sp_shape, dtypes.int32) |
| 475 | segment_ids = segment_ids * sp_shape[1] + column_ids |
| 476 | total_size = sp_shape[0] * sp_shape[1] |
| 477 | embeddings = math_ops.unsorted_segment_sum(embeddings, segment_ids, total_size) |
| 478 | embeddings = array_ops.reshape( |
| 479 | embeddings, [sp_shape[0], sp_shape[1] * array_ops.shape(embeddings)[-1]]) |
| 480 | return embeddings |
| 481 | |
| 482 | |
| 483 | @tf_export(v1=["nn.embedding_lookup_sparse"]) |
no test coverage detected