curobo.rollout package¶
This module contains rollout classes
Subpackages¶
- curobo.rollout.cost package
- Submodules
- curobo.rollout.cost.bound_cost module
- curobo.rollout.cost.cost_base module
- curobo.rollout.cost.dist_cost module
- curobo.rollout.cost.manipulability_cost module
- curobo.rollout.cost.pose_cost module
- curobo.rollout.cost.primitive_collision_cost module
- curobo.rollout.cost.projected_dist_cost module
- curobo.rollout.cost.self_collision_cost module
- curobo.rollout.cost.stop_cost module
- curobo.rollout.cost.straight_line_cost module
- curobo.rollout.cost.zero_cost module
- Submodules
- curobo.rollout.dynamics_model package
- Submodules
- curobo.rollout.dynamics_model.integration_utils module
build_clique_matrixbuild_fd_matrixbuild_int_matrixbuild_start_state_masktensor_step_jerkeuler_integratetensor_step_accjit_tensor_step_pos_clique_contiguousjit_tensor_step_pos_cliquejit_backward_pos_cliquejit_backward_pos_clique_contiguousCliqueTensorStepCliqueTensorStepKernelCliqueTensorStepIdxKernelCliqueTensorStepCentralDifferenceKernelCliqueTensorStepIdxCentralDifferenceKernelCliqueTensorStepCoalesceKernelAccelerationTensorStepKernelAccelerationTensorStepIdxKerneltensor_step_pos_cliquestep_acc_semi_eulertensor_step_acc_semi_eulertensor_step_veltensor_step_postensor_step_pos_iktensor_linspacesum_matrixinterpolate_kernelaction_interpolate_kernel
- curobo.rollout.dynamics_model.kinematic_model module
- curobo.rollout.dynamics_model.model_base module
- curobo.rollout.dynamics_model.tensor_step module
- curobo.rollout.dynamics_model.integration_utils module
- Submodules
Submodules¶
- curobo.rollout.arm_base module
ArmCostConfigArmBaseConfigArmBaseConfig.model_cfgArmBaseConfig.cost_cfgArmBaseConfig.constraint_cfgArmBaseConfig.convergence_cfgArmBaseConfig.world_coll_checkerArmBaseConfig.model_from_dictArmBaseConfig.cost_from_dictArmBaseConfig.world_coll_checker_from_dictArmBaseConfig.from_dictArmBaseConfig.sampler_seedArmBaseConfig.sum_horizonArmBaseConfig.tensor_args
ArmBaseArmBase._init_after_config_loadArmBase.cost_fnArmBase.constraint_fnArmBase.get_metricsArmBase.get_metrics_cuda_graphArmBase.convergence_fnArmBase._get_augmented_stateArmBase.compute_kinematicsArmBase.rollout_constraintArmBase.rollout_constraint_cuda_graphArmBase.rollout_fnArmBase.update_paramsArmBase.get_ee_poseArmBase.current_costArmBase.filter_robot_stateArmBase.get_robot_commandArmBase.resetArmBase.d_actionArmBase.action_bound_lowsArmBase.action_bound_highsArmBase.state_boundsArmBase.dtArmBase.horizonArmBase.action_horizonArmBase.get_init_action_seqArmBase.reset_shapeArmBase.reset_cuda_graphArmBase.get_action_from_stateArmBase.get_state_from_actionArmBase.kinematicsArmBase.cspace_configArmBase.get_full_dof_from_solutionArmBase.joint_namesArmBase.retract_stateArmBase.update_traj_dtArmBase.update_cost_dtArmBase.action_boundsArmBase.break_cuda_graphArmBase.constraint_cfgArmBase.convergence_cfgArmBase.cost_cfgArmBase.cost_from_dictArmBase.cuda_graph_instanceArmBase.from_dictArmBase.model_cfgArmBase.model_from_dictArmBase.reset_seedArmBase.sample_random_actionsArmBase.sampler_seedArmBase.sum_horizonArmBase.update_start_stateArmBase.world_coll_checkerArmBase.world_coll_checker_from_dictArmBase.tensor_args
- curobo.rollout.arm_reacher module
ArmReacherMetricsArmReacherMetrics.cspace_errorArmReacherMetrics.position_errorArmReacherMetrics.rotation_errorArmReacherMetrics.pose_errorArmReacherMetrics.goalset_indexArmReacherMetrics.null_space_errorArmReacherMetrics.cloneArmReacherMetrics._abc_implArmReacherMetrics._is_protocolArmReacherMetrics.constraintArmReacherMetrics.costArmReacherMetrics.countArmReacherMetrics.feasibleArmReacherMetrics.indexArmReacherMetrics.state
ArmReacherCostConfigArmReacherCostConfig.pose_cfgArmReacherCostConfig.cspace_cfgArmReacherCostConfig.straight_line_cfgArmReacherCostConfig.zero_acc_cfgArmReacherCostConfig.zero_vel_cfgArmReacherCostConfig.zero_jerk_cfgArmReacherCostConfig.link_pose_cfgArmReacherCostConfig._get_base_keysArmReacherCostConfig.from_dictArmReacherCostConfig._get_formatted_dictArmReacherCostConfig.bound_cfgArmReacherCostConfig.manipulability_cfgArmReacherCostConfig.null_space_cfgArmReacherCostConfig.primitive_collision_cfgArmReacherCostConfig.self_collision_cfgArmReacherCostConfig.stop_cfg
ArmReacherConfigArmReacherConfig.cost_cfgArmReacherConfig.constraint_cfgArmReacherConfig.convergence_cfgArmReacherConfig.cost_from_dictArmReacherConfig.from_dictArmReacherConfig.model_cfgArmReacherConfig.model_from_dictArmReacherConfig.sampler_seedArmReacherConfig.sum_horizonArmReacherConfig.world_coll_checkerArmReacherConfig.world_coll_checker_from_dictArmReacherConfig.tensor_args
_compute_g_dist_jitArmReacherArmReacher.cost_fnArmReacher.convergence_fnArmReacher.update_paramsArmReacher.enable_pose_costArmReacher._get_augmented_stateArmReacher._init_after_config_loadArmReacher.action_bound_highsArmReacher.action_bound_lowsArmReacher.action_boundsArmReacher.action_horizonArmReacher.break_cuda_graphArmReacher.compute_kinematicsArmReacher.constraint_cfgArmReacher.constraint_fnArmReacher.convergence_cfgArmReacher.cost_cfgArmReacher.cost_from_dictArmReacher.cspace_configArmReacher.cuda_graph_instanceArmReacher.current_costArmReacher.d_actionArmReacher.dtArmReacher.enable_cspace_costArmReacher.filter_robot_stateArmReacher.from_dictArmReacher.get_action_from_stateArmReacher.get_ee_poseArmReacher.get_full_dof_from_solutionArmReacher.get_init_action_seqArmReacher.get_metricsArmReacher.get_metrics_cuda_graphArmReacher.get_robot_commandArmReacher.get_state_from_actionArmReacher.horizonArmReacher.joint_namesArmReacher.kinematicsArmReacher.model_cfgArmReacher.model_from_dictArmReacher.resetArmReacher.reset_cuda_graphArmReacher.reset_seedArmReacher.reset_shapeArmReacher.retract_stateArmReacher.rollout_constraintArmReacher.rollout_constraint_cuda_graphArmReacher.rollout_fnArmReacher.sample_random_actionsArmReacher.sampler_seedArmReacher.state_boundsArmReacher.sum_horizonArmReacher.update_cost_dtArmReacher.update_start_stateArmReacher.update_traj_dtArmReacher.world_coll_checkerArmReacher.world_coll_checker_from_dictArmReacher.tensor_argsArmReacher.get_pose_costsArmReacher.update_pose_cost_metric
cat_sum_reachercat_sum_horizon_reacher
- curobo.rollout.rollout_base module
RolloutMetricsTrajectoryGoalGoal.nameGoal.goal_stateGoal.goal_poseGoal.links_goal_poseGoal.current_stateGoal.retract_stateGoal.batchGoal.batch_pose_idxGoal.batch_goal_state_idxGoal.batch_retract_state_idxGoal.batch_current_state_idxGoal.batch_enable_idxGoal.batch_world_idxGoal.update_batch_idx_buffersGoal.n_goalsetGoal._update_batch_sizeGoal.repeat_seedsGoal.cloneGoal._tensor_repeat_seedsGoal.apply_kernelGoal.toGoal.copy_Goal._copy_bufferGoal._copy_tensorGoal.get_batch_goal_stateGoal.create_index_buffersGoal.create_idxGoal._abc_implGoal._is_protocolGoal.countGoal.index
RolloutConfigRolloutBaseRolloutBase._init_after_config_loadRolloutBase.cost_fnRolloutBase.constraint_fnRolloutBase.convergence_fnRolloutBase.get_metricsRolloutBase.get_metrics_cuda_graphRolloutBase.rollout_fnRolloutBase.current_costRolloutBase.update_paramsRolloutBase.action_boundsRolloutBase.filter_robot_stateRolloutBase.get_robot_commandRolloutBase.reset_seedRolloutBase.resetRolloutBase.d_actionRolloutBase.action_bound_lowsRolloutBase.action_bound_highsRolloutBase.dtRolloutBase.horizonRolloutBase.action_horizonRolloutBase.update_start_stateRolloutBase.get_init_action_seqRolloutBase.state_boundsRolloutBase.sample_random_actionsRolloutBase.rollout_constraintRolloutBase.reset_cuda_graphRolloutBase.reset_shapeRolloutBase.cuda_graph_instanceRolloutBase.get_action_from_stateRolloutBase.get_state_from_actionRolloutBase.cspace_configRolloutBase.get_full_dof_from_solutionRolloutBase.break_cuda_graph
tensor_repeat_seeds