print(f'edge_index = {data.edge_index.shape}')
print(data.edge_index)