import torch


def get_r1_src_target(k_in=2, k_out=1):
    """
    Returns the r1 experiment src and targets tensors
    """
    # torch.set_default_dtype(torch.float64) # works best in float64
    points_tens = torch.tensor(
        [[-k_in, 0, 0], [k_in, 0, 0]]).unsqueeze(0).float()
    target_points_tens = (
        torch.tensor([[0, -k_out, 0], [0, k_out, 0]]).unsqueeze(0).float()
    )
    return points_tens, target_points_tens