Skip to content
Snippets Groups Projects

Matmul rework

Merged Houssem ROUIS requested to merge (removed):matmul_rework into dev

The behavior of the reworked MatMul on the arguments in the following way.
- If both arguments are 2-D they are multiplied like conventional matrices.
Basic_MMM - If either argument is N-D, N > 2, it is treated as a stack of matrices residing in the last two indexes.
Exp: For the MatMul of tensor A of shape (3,3,2) with tensor B of shape (3,2,4), A is a stack of three 2D matrices each of shape (3,2), and B is a stack of 3 2D matrices, each of shape (2,4).

Edited by Houssem ROUIS

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading