ray.rllib.core.learner.learner.Learner._convert_batch_type

abstract Learner._convert_batch_type(batch: MultiAgentBatch) → MultiAgentBatch

Converts the elements of a MultiAgentBatch to Tensors on the correct device.

Parameters:

batch – The MultiAgentBatch object to convert.

Returns:

The resulting MultiAgentBatch with framework-specific tensor values placed on the correct device.