Why embedding_bag result is different between GPU and CPU impl

1

when indices and offset are all 2D,

CPU:

cpu1
cpu1923×641 35.5 KB

2

GPU :

gpu
gpu1343×441 13.5 KB

3

Could you post a minimal and executable code snippet by wrapping them into three backticks ``` so we could try to reproduce and debug the issue?