| 6 | |
| 7 | |
| 8 | class Graph(NamedTuple): |
| 9 | x: torch.FloatTensor |
| 10 | edge_index: torch.LongTensor |
| 11 | edge_weights: Optional[torch.FloatTensor] |
| 12 | |
| 13 | def unfold(self) -> Tuple[torch.FloatTensor, torch.LongTensor, Optional[torch.FloatTensor]]: |
| 14 | return self.x, self.edge_index, self.edge_weights |
| 15 | |
| 16 | |
| 17 | class Augmentor(ABC): |