PyTorch Scatter documentation

This package consists of a small extension library of highly optimized sparse update (scatter) operations for the use in PyTorch, which are missing in the main package. Scatter operations can be roughly described as reduce operations based on a given “group-index” tensor.

All included operations work on varying data types, are implemented both for CPU and GPU and include a backwards implementation.

Package reference

• Scatter Add
• Scatter Div
• Scatter Max
• Scatter Mean
• Scatter Min
• Scatter Mul
• Scatter Std
• Scatter Sub

Indices and tables

• Index
• Module Index