PyTorch Geometric (PyG)
1. **edge_index shape**: Must be `[2, num_edges]`, not `[num_edges, 2]`. Transpose if needed.
LLM Evaluation
Evaluated by: xiaomi/mimo-v2-flash:free
Last evaluated: May 17, 2026
Prompt Preview
---
name: torch-geometric
description: "Guide for building Graph Neural Networks with PyTorch Geometric (PyG). Use this skill whenever the user asks about graph neural networks, GNNs, node classification, link prediction, graph classification, message passing networks, heterogeneous graphs, neighbor sampling, or any task involving torch_geometric / PyG. Also trigger when you see imports from torch_geometric, or the user mentions graph convolutions (GCN, GAT, GraphSAGE, GIN), graph data structure...Full prompt length: 17029 characters
Tools & Technologies
- python
- rest