Quick way to find out the mean of neighbors?

[jonathan-h-w]

Given node features and edge_index, is there a quick way to find out the mean of neighbors for each node? Expecting the output tensor to be the same shape as node features.

[EdisonLeeeee]

from torch_geometric.nn.aggr import MeanAggregation
row, col = edge_index
x_mean = MeanAggregation()(x[row], col)

[jonathan-h-w]

Thanks Edison! Should've looked into the docs more carefully before posting

[EdisonLeeeee]

It's okay. Glad to help you :)