tidy matrix display

Author: leej3

Diff for: 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}Shape: {list(item.shape)}"
         elif isinstance(item, dict):
             for key, value in item.items():
                 formatted_value = format_item(value, f"{key}: ")

Diff for: tests/ignite/test_utils.py

@@ -67,7 +67,7 @@ def test_convert_tensor():
             {"large_vector": torch.tensor(range(20))},
             ["large_vector: [0.0000, 1.0000, 2.0000, 3.0000, 4.0000, 5.0000, 6.0000, 7.0000, 8.0000, 9.0000, ...]"],
         ),
-        ({"large_matrix": torch.randn(5, 5)}, ["large_matrix: Shape: torch.Size([5, 5])"]),
+        ({"large_matrix": torch.randn(5, 5)}, ["large_matrix: Shape: [5, 5]"]),
         ({"empty": []}, ["empty: "]),
         ([], []),
         ({"none": None}, ["none: "]),