openvino.runtime.opset5.embedding_bag_packed_sum

openvino.runtime.opset5.embedding_bag_packed_sum(emb_table: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], indices: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], per_sample_weights: Optional[Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]] = None, name: Optional[str] = None) openvino._pyopenvino.Node

Return an EmbeddingBagPackedSum 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.

  • per_sample_weights – Weights to be multiplied with embedding table.

  • name – Optional name for output node.

Returns

EmbeddingBagPackedSum node