From b7825b08311e8dcbffe0ac5bf99ee37fef44119d Mon Sep 17 00:00:00 2001 From: franksim Date: Tue, 14 Oct 2025 09:37:51 +0000 Subject: [PATCH] assignment-a1: adds add_matrix_vector and avg_color --- mmp/a1/tensors.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/mmp/a1/tensors.py b/mmp/a1/tensors.py index 4574de1..0c7c329 100644 --- a/mmp/a1/tensors.py +++ b/mmp/a1/tensors.py @@ -1,10 +1,17 @@ import torch +from PIL import Image +from torchvision.transforms import ToTensor + def avg_color(img: torch.Tensor): - raise NotImplementedError() + result = img.mean(dim=(1, 2)).tolist() + return tuple(result) + def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float): raise NotImplementedError() + def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor): - raise NotImplementedError() + return matrix.add(vector) +