ray.rllib.policy.sample_batch.SampleBatch.to_device

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

TODO: 将批处理转移到指定设备作为框架张量。