tidy matrix display

pytorch · Jun 22, 2024 · 1e37dce · 1e37dce
1 parent 2f5833c
commit 1e37dce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/utils.py b/ignite/utils.py
@@ -140,7 +140,7 @@ def format_item(item: Any, prefix: str = "") -> Optional[str]:
             elif item.dim() == 1:
                 return f"{prefix}[" + ", ".join(f"{x.item():.4f}" for x in item) + "]"
             else:
-                return f"{prefix}Shape: {item.shape}"
+                return f"{prefix} {list(item.shape)}"
         elif isinstance(item, dict):
             for key, value in item.items():
                 formatted_value = format_item(value, f"{key}: ")