2bd6dff4cacf1e25989f04cf8a243c3c06ab1269,torch_geometric/data/batch.py,Batch,from_data_list,#Any#Any#,22

Before Change

                cumsum[key] += data.__inc__(key, item)
                batch[key].append(item)

            for key in follow_batch:
                item = torch.full((size, ), i, dtype=torch.long)
                batch["{}_batch".format(key)].append(item)

            num_nodes = data.num_nodes
            if num_nodes is not None:
                item = torch.full((num_nodes, ), i, dtype=torch.long)
                batch.batch.append(item)

After Change

        for i, data in enumerate(data_list):
            for key in data.keys:
                item = data[key] + cumsum[key]
                if torch.is_tensor(data[key]):
                    size = data[key].size(data.__cat_dim__(key, data[key]))
                else:
                    size = 1
                batch.__slices__[key].append(size + batch.__slices__[key][-1])
                cumsum[key] += data.__inc__(key, item)
                batch[key].append(item)

Instances