Get shape string from model output.
(output: Any)
| 962 | |
| 963 | |
| 964 | def _output_shape_str(output: Any) -> str: |
| 965 | """Get shape string from model output.""" |
| 966 | try: |
| 967 | t = extract_tensor(output) |
| 968 | return str(list(t.shape)) |
| 969 | except Exception: |
| 970 | return "unknown" |
| 971 | |
| 972 | |
| 973 | # --------------------------------------------------------------------------- |
nothing calls this directly
no test coverage detected