ray.rllib.policy.sample_batch.MultiAgentBatch.to_device

MultiAgentBatch.to_device(device, framework='torch')

TODO: transfer batch to given device as framework tensor.