differentiable_robot_model/data_utils.py [118:141]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    T = int(n_data * dt)
    t = torch.linspace(0.0, T - 1, n_data)
    pi = torch.acos(torch.zeros(1)).item() * 2

    q = A * torch.sin(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs) + q_start
    qd = (
        2
        * pi
        * freq
        * A
        * torch.cos(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs)
    )
    qdd_des = (
        -((2 * pi * freq) ** 2)
        * A
        * torch.sin(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs)
    )

    q = q.to(device)
    qd = qd.to(device)
    qdd_des = qdd_des.to(device)
    torques = robot_model.compute_inverse_dynamics(
        q=q, qd=qd, qdd_des=qdd_des, include_gravity=True
    )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



differentiable_robot_model/data_utils.py [154:177]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    T = int(n_data * dt)
    t = torch.linspace(0.0, T - 1, n_data)
    pi = torch.acos(torch.zeros(1)).item() * 2

    q = A * torch.sin(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs) + q_start
    qd = (
        2
        * pi
        * freq
        * A
        * torch.cos(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs)
    )
    qdd_des = (
        -((2 * pi * freq) ** 2)
        * A
        * torch.sin(2 * pi * freq * t).reshape(n_data, 1).repeat(1, n_dofs)
    )

    q = q.to(device)
    qd = qd.to(device)
    qdd_des = qdd_des.to(device)
    torques = robot_model.compute_inverse_dynamics(
        q=q, qd=qd, qdd_des=qdd_des, include_gravity=True
    )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -