mmcm/utils/compute_util.py

from typing import List
import numpy as np


def weighted_sum(weights: List[float], datas: List[np.array]) -> np.array:
    """对矩阵列表按照权重列表加权求和

    Args:
        weights (list): 权重列表
        datas (list): 矩阵列表

    Returns:
        np.array: 加权求和后的矩阵
    """
    res = np.zeros(datas[0].shape)
    n_data = len(datas)
    for i in range(n_data):
        res += datas[i] * weights[i] / n_data
    return res