matrix.py 文件源码

def mix(w: T.FloatingPoint,
        x: T.FloatTensor,
        y: T.FloatTensor) -> None:
    """
    Compute a weighted average of two matrices (x and y) and return the result.
    Multilinear interpolation.

    Note:
        Modifies x in place.

    Args:
        w: The mixing coefficient (float or tensor) between 0 and 1.
        x: A tensor.
        y: A tensor:

    Returns:
        tensor = w * x + (1-w) * y

    """
    return torch.add(torch.mul(x, w), torch.mul(1-w, y))