akankshakshurai/cuda-error-resolution-analysis huggingface dataset:
How GNN layers can know which graph in a batch has the given node & node feature?

username: cramraj8
cooked: <p>When we pass input as node features (x) and edge index (edge_index) to pytorch_geometric layer (e.g. GATConv), I am worried whether the layer can differentiate which batch sample the given node elements belong to.</p>
<p>x follows the shape [num of nodes, feature size] and edge_index follows shape [2, num of edges]. However, these 2 do not have the given information to know which input graph of batch size 32 have given node feature in the x.</p>
<p>Anyone can clarify on this ?</p>