(seg_out)
| 421 | |
| 422 | |
| 423 | def _flatten_struct(seg_out): |
| 424 | dict_keys = None |
| 425 | seg_probs: tuple[torch.Tensor, ...] |
| 426 | if isinstance(seg_out, torch.Tensor): |
| 427 | seg_probs = (seg_out,) |
| 428 | elif isinstance(seg_out, Mapping): |
| 429 | dict_keys = sorted(seg_out.keys()) # track predictor's output keys |
| 430 | seg_probs = tuple(seg_out[k] for k in dict_keys) |
| 431 | else: |
| 432 | seg_probs = ensure_tuple(seg_out) |
| 433 | return dict_keys, seg_probs |
| 434 | |
| 435 | |
| 436 | def _pack_struct(seg_out, dict_keys=None): |
no test coverage detected
searching dependent graphs…