Torch scatter reduce. .

Torch scatter reduce .