mmp_wise2526_franksim/mmp/a1/tensors.py

import torch
from PIL import Image
from torchvision.transforms import ToTensor


def avg_color(img: torch.Tensor):
    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):
    return matrix.add(vector)