From fe329b636d71c0c5099a618426360c91cfd63fe9 Mon Sep 17 00:00:00 2001 From: franksim Date: Thu, 16 Oct 2025 14:17:54 +0000 Subject: [PATCH] assignment-a1: fix --- mmp/a1/tensors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmp/a1/tensors.py b/mmp/a1/tensors.py index f96fbac..3d87203 100644 --- a/mmp/a1/tensors.py +++ b/mmp/a1/tensors.py @@ -2,7 +2,7 @@ import torch def avg_color(img: torch.Tensor): - return img.mean(dim=(1, 2)).tolist() + return img.mean(dim=(1, 2)) def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float):