fix tensor shape to match expected (B=1, C=3, H=1, W=2) format

Signed-off-by: ytl0623 <[email protected]>

Author: ytl0623

tests/losses/test_unified_focal_loss.py

@@ -48,13 +48,8 @@
             "include_background": True,
         },
         {
-            # Logits:
-            # Pixel 1: Class 0 is target -> [10, -10, -10]
-            # Pixel 2: Class 2 is target -> [-10, -10, 10]
-            "y_pred": torch.tensor(
-                [[[[logit_pos, logit_neg], [logit_neg, logit_neg], [logit_neg, logit_pos]]]]  # Ch 0  # Ch 1  # Ch 2
-            ),
-            "y_true": torch.tensor([[[[1.0, 0.0], [0.0, 0.0], [0.0, 1.0]]]]),  # Ch 0 (Background)  # Ch 1  # Ch 2
+            "y_pred": torch.tensor([[[[logit_pos, logit_neg]], [[logit_neg, logit_neg]], [[logit_neg, logit_pos]]]]),
+            "y_true": torch.tensor([[[[1.0, 0.0]], [[0.0, 0.0]], [[0.0, 1.0]]]]),
         },
         0.0,
     ],