batch (int tensor, (n_nodes)): The batch each new node
is part of.
x = x / unpool_info.new_nodes_edge_scores[:, None]
new_x = x[unpool_info.old_to_new_node_idx]
return new_x, unpool_info.edge_index, unpool_info.batch
def __repr__(self):
After Change
batch (LongTensor): The new batch vector.
new_x = x / unpool_info.new_edge_score.view(-1, 1)
new_x = new_x[unpool_info.cluster]
return new_x, unpool_info.edge_index, unpool_info.batch
def __repr__(self):