/dgl_batchbox

This is Pytorch repository for the common operations in DGL in a batch-wised manner.

Primary LanguagePythonMIT LicenseMIT

dgl_batchbox

This is a Pytorch repository for the common operations in DGL in a batch-wised manner. There are several operations that are currently supported as below:

Normalization

  1. PairNorm

Pooling

  1. SAGPool

Graph Operations

Edge-based

  1. Adding self-loop
  2. Convert to bidirected graph

Installation

pip install -i https://test.pypi.org/simple/ dgl-batchbox

Notes

  • This package currenly supports homogenous operations only.
  • New operations will be added as on-going basis. However, contributions are greatly appreciated.