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

abstract Learner._convert_batch_type(batch: ray.rllib.policy.sample_batch.MultiAgentBatch) → ray.rllib.policy.sample_batch.MultiAgentBatch[source]#

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.