PinMemoryTransform

class torchrl.envs.transforms.PinMemoryTransform

在 tensordict 上调用 pin_memory 以便于在 CUDA 设备上进行写入操作。

forward(tensordict: TensorDictBase) → TensorDictBase

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