├── .coveragerc ├── .github └── workflows │ ├── building.yml │ ├── cuda │ ├── cu101-Linux-env.sh │ ├── cu101-Linux.sh │ ├── cu101-Windows-env.sh │ ├── cu101-Windows.sh │ ├── cu102-Linux-env.sh │ ├── cu102-Linux.sh │ ├── cu102-Windows-env.sh │ ├── cu102-Windows.sh │ ├── cu111-Linux-env.sh │ ├── cu111-Linux.sh │ ├── cu111-Windows-env.sh │ ├── cu111-Windows.sh │ ├── cu113-Linux-env.sh │ ├── cu113-Linux.sh │ ├── cu113-Windows-env.sh │ ├── cu113-Windows.sh │ ├── cu115-Linux-env.sh │ ├── cu115-Linux.sh │ ├── cu115-Windows-env.sh │ ├── cu115-Windows.sh │ ├── cu116-Linux-env.sh │ ├── cu116-Linux.sh │ ├── cu116-Windows-env.sh │ ├── cu116-Windows.sh │ ├── cu117-Linux-env.sh │ ├── cu117-Linux.sh │ ├── cu117-Windows-env.sh │ ├── cu117-Windows.sh │ ├── cu118-Linux-env.sh │ ├── cu118-Linux.sh │ ├── cu118-Windows-env.sh │ ├── cu118-Windows.sh │ ├── cu121-Linux-env.sh │ ├── cu121-Linux.sh │ ├── cu121-Windows-env.sh │ ├── cu121-Windows.sh │ ├── cu124-Linux-env.sh │ ├── cu124-Linux.sh │ ├── cu124-Windows-env.sh │ ├── cu124-Windows.sh │ ├── cu126-Linux-env.sh │ ├── cu126-Linux.sh │ ├── cu126-Windows-env.sh │ ├── cu126-Windows.sh │ ├── cu128-Linux-env.sh │ ├── cu128-Linux.sh │ ├── cu128-Windows-env.sh │ ├── cu128-Windows.sh │ ├── cu129-Linux-env.sh │ ├── cu129-Linux.sh │ ├── cu129-Windows-env.sh │ └── cu129-Windows.sh │ ├── linting.yml │ ├── stale.yml │ └── testing.yml ├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── MANIFEST.in ├── README.md ├── benchmark ├── .gitignore ├── gather.py └── scatter_segment.py ├── cmake └── TorchScatterConfig.cmake.in ├── csrc ├── cpu │ ├── index_info.h │ ├── reducer.h │ ├── scatter_cpu.cpp │ ├── scatter_cpu.h │ ├── segment_coo_cpu.cpp │ ├── segment_coo_cpu.h │ ├── segment_csr_cpu.cpp │ ├── segment_csr_cpu.h │ └── utils.h ├── cuda │ ├── atomics.cuh │ ├── index_info.cuh │ ├── reducer.cuh │ ├── scatter_cuda.cu │ ├── scatter_cuda.h │ ├── segment_coo_cuda.cu │ ├── segment_coo_cuda.h │ ├── segment_csr_cuda.cu │ ├── segment_csr_cuda.h │ └── utils.cuh ├── extensions.h ├── macros.h ├── scatter.cpp ├── scatter.h ├── segment_coo.cpp ├── segment_csr.cpp ├── utils.h └── version.cpp ├── docs ├── .nojekyll ├── Makefile ├── index.html ├── requirements.txt └── source │ ├── _figures │ ├── add.svg │ ├── add.tex │ ├── build.sh │ ├── div.svg │ ├── div.tex │ ├── max.svg │ ├── max.tex │ ├── mean.svg │ ├── mean.tex │ ├── min.svg │ ├── min.tex │ ├── mul.svg │ ├── mul.tex │ ├── segment_coo.svg │ ├── segment_coo.tex │ ├── std.svg │ ├── std.tex │ ├── sub.svg │ ├── sub.tex │ └── template.tex │ ├── conf.py │ ├── functions │ ├── composite.rst │ ├── scatter.rst │ ├── segment_coo.rst │ └── segment_csr.rst │ └── index.rst ├── pyproject.toml ├── readthedocs.yml ├── setup.cfg ├── setup.py ├── test ├── composite │ ├── test_logsumexp.py │ ├── test_softmax.py │ └── test_std.py ├── test_broadcasting.py ├── test_gather.py ├── test_multi_gpu.py ├── test_scatter.py ├── test_segment.py └── test_zero_tensors.py └── torch_scatter ├── __init__.py ├── composite ├── __init__.py ├── logsumexp.py ├── softmax.py └── std.py ├── placeholder.py ├── scatter.py ├── segment_coo.py ├── segment_csr.py ├── testing.py └── utils.py /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.coveragerc -------------------------------------------------------------------------------- /.github/workflows/building.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/building.yml -------------------------------------------------------------------------------- /.github/workflows/cuda/cu101-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu101-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu101-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu101-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu101-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu101-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu101-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu101-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu102-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu102-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu102-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu102-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu102-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu102-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu102-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu102-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu111-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu111-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu111-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu111-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu111-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu111-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu111-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu111-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu113-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu113-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu113-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu113-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu113-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu113-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu113-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu113-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu115-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu115-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu115-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu115-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu115-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu115-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu115-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu115-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu116-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu116-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu116-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu116-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu116-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu116-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu116-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu116-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu117-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu117-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu117-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu117-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu117-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu117-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu117-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu117-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu118-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu118-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu118-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu118-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu118-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu118-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu118-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu118-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu121-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu121-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu121-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu121-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu121-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu121-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu121-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu121-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu124-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu124-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu124-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu124-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu124-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu124-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu124-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu124-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu126-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu126-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu126-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu126-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu126-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu126-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu126-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu126-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu128-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu128-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu128-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu128-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu128-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu128-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu128-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu128-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu129-Linux-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu129-Linux-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu129-Linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu129-Linux.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu129-Windows-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu129-Windows-env.sh -------------------------------------------------------------------------------- /.github/workflows/cuda/cu129-Windows.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/cuda/cu129-Windows.sh -------------------------------------------------------------------------------- /.github/workflows/linting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/linting.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/stale.yml -------------------------------------------------------------------------------- /.github/workflows/testing.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.github/workflows/testing.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/.gitignore: -------------------------------------------------------------------------------- 1 | *.mat 2 | *.tmp 3 | -------------------------------------------------------------------------------- /benchmark/gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/benchmark/gather.py -------------------------------------------------------------------------------- /benchmark/scatter_segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/benchmark/scatter_segment.py -------------------------------------------------------------------------------- /cmake/TorchScatterConfig.cmake.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/cmake/TorchScatterConfig.cmake.in -------------------------------------------------------------------------------- /csrc/cpu/index_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/index_info.h -------------------------------------------------------------------------------- /csrc/cpu/reducer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/reducer.h -------------------------------------------------------------------------------- /csrc/cpu/scatter_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/scatter_cpu.cpp -------------------------------------------------------------------------------- /csrc/cpu/scatter_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/scatter_cpu.h -------------------------------------------------------------------------------- /csrc/cpu/segment_coo_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/segment_coo_cpu.cpp -------------------------------------------------------------------------------- /csrc/cpu/segment_coo_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/segment_coo_cpu.h -------------------------------------------------------------------------------- /csrc/cpu/segment_csr_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/segment_csr_cpu.cpp -------------------------------------------------------------------------------- /csrc/cpu/segment_csr_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/segment_csr_cpu.h -------------------------------------------------------------------------------- /csrc/cpu/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cpu/utils.h -------------------------------------------------------------------------------- /csrc/cuda/atomics.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/atomics.cuh -------------------------------------------------------------------------------- /csrc/cuda/index_info.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/index_info.cuh -------------------------------------------------------------------------------- /csrc/cuda/reducer.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/reducer.cuh -------------------------------------------------------------------------------- /csrc/cuda/scatter_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/scatter_cuda.cu -------------------------------------------------------------------------------- /csrc/cuda/scatter_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/scatter_cuda.h -------------------------------------------------------------------------------- /csrc/cuda/segment_coo_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/segment_coo_cuda.cu -------------------------------------------------------------------------------- /csrc/cuda/segment_coo_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/segment_coo_cuda.h -------------------------------------------------------------------------------- /csrc/cuda/segment_csr_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/segment_csr_cuda.cu -------------------------------------------------------------------------------- /csrc/cuda/segment_csr_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/segment_csr_cuda.h -------------------------------------------------------------------------------- /csrc/cuda/utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/cuda/utils.cuh -------------------------------------------------------------------------------- /csrc/extensions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/extensions.h -------------------------------------------------------------------------------- /csrc/macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/macros.h -------------------------------------------------------------------------------- /csrc/scatter.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/scatter.cpp -------------------------------------------------------------------------------- /csrc/scatter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/scatter.h -------------------------------------------------------------------------------- /csrc/segment_coo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/segment_coo.cpp -------------------------------------------------------------------------------- /csrc/segment_csr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/segment_csr.cpp -------------------------------------------------------------------------------- /csrc/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/utils.h -------------------------------------------------------------------------------- /csrc/version.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/csrc/version.cpp -------------------------------------------------------------------------------- /docs/.nojekyll: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_figures/add.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/add.svg -------------------------------------------------------------------------------- /docs/source/_figures/add.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/add.tex -------------------------------------------------------------------------------- /docs/source/_figures/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/build.sh -------------------------------------------------------------------------------- /docs/source/_figures/div.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/div.svg -------------------------------------------------------------------------------- /docs/source/_figures/div.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/div.tex -------------------------------------------------------------------------------- /docs/source/_figures/max.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/max.svg -------------------------------------------------------------------------------- /docs/source/_figures/max.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/max.tex -------------------------------------------------------------------------------- /docs/source/_figures/mean.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/mean.svg -------------------------------------------------------------------------------- /docs/source/_figures/mean.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/mean.tex -------------------------------------------------------------------------------- /docs/source/_figures/min.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/min.svg -------------------------------------------------------------------------------- /docs/source/_figures/min.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/min.tex -------------------------------------------------------------------------------- /docs/source/_figures/mul.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/mul.svg -------------------------------------------------------------------------------- /docs/source/_figures/mul.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/mul.tex -------------------------------------------------------------------------------- /docs/source/_figures/segment_coo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/segment_coo.svg -------------------------------------------------------------------------------- /docs/source/_figures/segment_coo.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/segment_coo.tex -------------------------------------------------------------------------------- /docs/source/_figures/std.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/std.svg -------------------------------------------------------------------------------- /docs/source/_figures/std.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/std.tex -------------------------------------------------------------------------------- /docs/source/_figures/sub.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/sub.svg -------------------------------------------------------------------------------- /docs/source/_figures/sub.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/sub.tex -------------------------------------------------------------------------------- /docs/source/_figures/template.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/_figures/template.tex -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/functions/composite.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/functions/composite.rst -------------------------------------------------------------------------------- /docs/source/functions/scatter.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/functions/scatter.rst -------------------------------------------------------------------------------- /docs/source/functions/segment_coo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/functions/segment_coo.rst -------------------------------------------------------------------------------- /docs/source/functions/segment_csr.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/functions/segment_csr.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/readthedocs.yml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/setup.py -------------------------------------------------------------------------------- /test/composite/test_logsumexp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/composite/test_logsumexp.py -------------------------------------------------------------------------------- /test/composite/test_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/composite/test_softmax.py -------------------------------------------------------------------------------- /test/composite/test_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/composite/test_std.py -------------------------------------------------------------------------------- /test/test_broadcasting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_broadcasting.py -------------------------------------------------------------------------------- /test/test_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_gather.py -------------------------------------------------------------------------------- /test/test_multi_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_multi_gpu.py -------------------------------------------------------------------------------- /test/test_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_scatter.py -------------------------------------------------------------------------------- /test/test_segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_segment.py -------------------------------------------------------------------------------- /test/test_zero_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/test/test_zero_tensors.py -------------------------------------------------------------------------------- /torch_scatter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/__init__.py -------------------------------------------------------------------------------- /torch_scatter/composite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/composite/__init__.py -------------------------------------------------------------------------------- /torch_scatter/composite/logsumexp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/composite/logsumexp.py -------------------------------------------------------------------------------- /torch_scatter/composite/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/composite/softmax.py -------------------------------------------------------------------------------- /torch_scatter/composite/std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/composite/std.py -------------------------------------------------------------------------------- /torch_scatter/placeholder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/placeholder.py -------------------------------------------------------------------------------- /torch_scatter/scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/scatter.py -------------------------------------------------------------------------------- /torch_scatter/segment_coo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/segment_coo.py -------------------------------------------------------------------------------- /torch_scatter/segment_csr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/segment_csr.py -------------------------------------------------------------------------------- /torch_scatter/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/testing.py -------------------------------------------------------------------------------- /torch_scatter/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rusty1s/pytorch_scatter/HEAD/torch_scatter/utils.py --------------------------------------------------------------------------------