ngraph.opset3.ops.embedding_bag_packed_sum¶
-
ngraph.opset3.ops.
embedding_bag_packed_sum
(emb_table: Union[_pyngraph.Node, int, float, numpy.ndarray], indices: Union[_pyngraph.Node, int, float, numpy.ndarray], per_sample_weights: Optional[Union[_pyngraph.Node, int, float, numpy.ndarray]] = None, name: Optional[str] = None) → _pyngraph.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