Skip to content

Commit

Permalink
Formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
mmikolajcz committed Jun 19, 2024
1 parent 624b722 commit 36f45d5
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,9 @@ EmbeddingBagPacked

Operation EmbeddingBagPacked is an implementation of ``torch.nn.EmbeddingBag`` with indices input being 2D tensor of shape ``[batch, indices_per_bag]``.
Operation is equivalent to *gather_op = Gather(emb_table, indices, axis=0)* followed by reduction:
* *sum* - *ReduceSum(Multiply(gather_op, Unsqueeze(per_sample_weights, -1)), axis=1)*,
* *mean* - *ReduceMean(gather_op, axis=1)*.

* *sum* - *ReduceSum(Multiply(gather_op, Unsqueeze(per_sample_weights, -1)), axis=1)*,
* *mean* - *ReduceMean(gather_op, axis=1)*.

**Attributes**:

Expand Down

0 comments on commit 36f45d5

Please sign in to comment.