ray.rllib.policy.sample_batch.SampleBatch.to_device

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

TODO: transfer batch to given device as framework tensor.