9e2a150b630e08a7037c1fd19e01cd5206e84a24,torch_geometric/transforms/grid_sampling.py,GridSampling,__call__,#GridSampling#Any#,22

Before Change

            if torch.is_tensor(item) and item.size(0) == num_nodes:
                if key == "y":
                    one_hot = torch.zeros((item.shape[0], self._num_classes))\
                        .scatter(1, item.unsqueeze(-1), 1)

                    aggr_labels = scatter_add(one_hot, pool, dim=0)

After Change

            if torch.is_tensor(item) and item.size(0) == num_nodes:
                if key == "y":
                    item = F.one_hot(item)
                    item = scatter_add(item, cluster, dim=0)
                    data[key] = item.argmax(dim=-1)
                elif key == "batch":
                    data[key] = item[perm]

Instances