Modules§
- matmul
- The matmul reading of a
Tile:c.mma(a, b)treats the trailing two axes as therow × colmatrix, leading axes as a batch, and contractsK. - softmax
- The softmax reading of a
Tile:OnlineSoftmaxtakes a score where its holder keeps it, the ownership read off the score.