planner.experts.expert_sampler#

Module Contents#

class planner.experts.expert_sampler.ExpertSampler(plant: jacta.planner.dynamics.simulator_plant.SimulatorPlant, graph: jacta.planner.core.graph.Graph, params: jacta.planner.core.parameter_container.ParameterContainer)#
Parameters:

  • plant (jacta.planner.dynamics.simulator_plant.SimulatorPlant) –

  • graph (jacta.planner.core.graph.Graph) –

  • params (jacta.planner.core.parameter_container.ParameterContainer) –

callback(node_ids: torch.IntTensor) → torch.FloatTensor#
Parameters:

node_ids (torch.IntTensor) –

Return type:

torch.FloatTensor