(self)
| 94 | self.device = device |
| 95 | |
| 96 | def __len__(self): |
| 97 | return len(self.occ) |
| 98 | |
| 99 | def __getitem__(self, idx): # occ: batch, seq, node |
| 100 | output_occ = torch.transpose(self.occ[idx, :, :], 0, 1).to(self.device) |
nothing calls this directly
no outgoing calls
no test coverage detected