From 1e37dcecc29481f3f80714401cdc484dd4752a19 Mon Sep 17 00:00:00 2001 From: leej3 Date: Sat, 22 Jun 2024 19:21:45 +0100 Subject: [PATCH] tidy matrix display --- ignite/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ignite/utils.py b/ignite/utils.py index c45c49a24497..ff9f44afc4c8 100644 --- 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}: ")