Function modulate

eval_code/recons/models/vggt/heads/camera_head.py:157–162

Modulate the input tensor using scaling and shifting parameters.

(x: torch.Tensor, shift: torch.Tensor, scale: torch.Tensor)

def modulate(x: torch.Tensor, shift: torch.Tensor, scale: torch.Tensor) -> torch.Tensor:
    """
    Modulate the input tensor using scaling and shifting parameters.
    """
    # modified from https://github.com/facebookresearch/DiT/blob/796c29e532f47bba17c5b9c5eb39b9354b8b7c64/models.py#L19
    return x * (1 + scale) + shift