assignment-a1: adds add_matrix_vector and avg_color

This commit is contained in:
franksim
2025-10-14 09:37:51 +00:00
parent 89faa2649b
commit b7825b0831

View File

@@ -1,10 +1,17 @@
import torch import torch
from PIL import Image
from torchvision.transforms import ToTensor
def avg_color(img: torch.Tensor): 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): def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float):
raise NotImplementedError() raise NotImplementedError()
def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor): def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor):
raise NotImplementedError() return matrix.add(vector)