openvino.runtime.opset14.embedding_segments_sum#

openvino.runtime.opset14.embedding_segments_sum(emb_table: Node, indices: Node | int | float | ndarray, segment_ids: Node | int | float | ndarray, num_segments: Node | int | float | ndarray | None = None, default_index: Node | int | float | ndarray | None = None, per_sample_weights: Node | int | float | ndarray | None = None, name: str | None = None) → Node#

Return an EmbeddingSegmentsSum node.

EmbeddingSegmentsSum constructs an output tensor by replacing every index in a given input tensor with a row (from the weights matrix) at that index

Parameters:

  • emb_table – Tensor containing the embedding lookup table.

  • indices – Tensor with indices.

  • segment_ids – Tensor with indices into the output Tensor

  • num_segments – Tensor with number of segments.

  • default_index – Scalar containing default index in embedding table to fill empty bags.

  • per_sample_weights – Weights to be multiplied with embedding table.

  • name – Optional name for output node.

Returns:

EmbeddingSegmentsSum node