ray.rllib.policy.sample_batch.SampleBatch.to_device#

SampleBatch.to_device(device, framework='torch')[source]#

TODO: transfer batch to given device as framework tensor.