• English
English

Hauptnavigation

PyTorch Geometric

Beschreibung:

PyTorch Geometric (PyG) ist eine auf PyTorch basierende Bibliothek für das tiefe Lernen auf irregulären Eingabedaten wie Graphen, Punktwolken oder Mannigfaltikeiten. Neben allgemeinen Datenstrukturen und Verarbeitungsmethoden enthält PyG eine Vielzahl von kürzlich veröffentlichten Methoden aus den Bereichen des relationalen Lernens und der 3D-Datenverarbeitung. PyG erreicht einen hohen Datendurchsatz durch die Nutzung von GPUs, durch die Bereitstellung dedizierter CUDA-Kernel und durch die effiziente Behandlung von Mini-Batches für Eingabedaten beliebiger Größe.

Link:

https://github.com/rusty1s/pytorch_geometric

Authors:

Fey, Matthias
Lenssen, Jan Eric

Projects:

SFB876-A6
SFB876-B2