Convert NumPy sample to PyTorch tensors.
(sample)
| 250 | |
| 251 | |
| 252 | def inplace_to_torch(sample): |
| 253 | """Convert NumPy sample to PyTorch tensors.""" |
| 254 | if sample is None: |
| 255 | return None |
| 256 | sample["features"] = {k: torch.FloatTensor(v) for k, v in sample["features"].items()} |
| 257 | sample["indexer"] = {k: torch.LongTensor(v) for k, v in sample["indexer"].items()} |
| 258 | if "labels" in sample.keys(): |
| 259 | sample["labels"] = {k: torch.FloatTensor(v) for k, v in sample["labels"].items()} |
| 260 | return sample |
| 261 | |
| 262 | |
| 263 | def distance_to_gaussian_contact_logits( |
no outgoing calls
no test coverage detected