3238295386c5c5e9d39ed770e92a54326430b8ae,torch_geometric/visualization/mnist_graph.py,,,#,11

Before Change

adj = grid(torch.Size([28, 28]), connectivity=8)
position = grid_position(torch.Size([28, 28]))
input = image.flatten() / 255.0

// image = image_graph(image, input, adj, position, scale, offset)
// image = image_graph(image, grid_input, grid_adj, grid_position, scale,
//                     grid_offset)

After Change

position += scale * offset * rescale

adjs, positions, perm = graclus(adj, position, level=2)
adj, position = adjs[2], positions[2]
input = perm_input(input, perm)
input = input.view(-1, 4).sum(dim=1)

index = adj._indices().t()

Instances