PinMemoryTransform

class torchrl.envs.transforms.PinMemoryTransform

在 tensordict 上调用 pin_memory 以便于写入 CUDA 设备。

forward(tensordict: TensorDictBase) → TensorDictBase

读取输入 tensordict，并对选定的键应用转换。