946352c486f5de2c1174a9403bde50ec11db5315,torch_geometric/datasets/faust.py,FAUST,__getitem__,#FAUST#Any#,55

Before Change

        vertices = self.vertices[index]
        edges = self.edges[index]
        data = (vertices, edges)

        // TODO: return adj with ones as edge weights?

        if self.correspondence:

After Change

    def __getitem__(self, i):
        position = self.position[i]
        index = self.index[:, i]
        weight = torch.FloatTensor(index.size(1)).fill_(1)
        input = torch.FloatTensor(position.size(0)).fill_(1)
        adj = torch.sparse.FloatTensor(index, weight, torch.Size([75, 75]))
        data = (input, adj, position)

        if self.correspondence:
            target = torch.arange(0, input.size(0)).long()

Instances