PinMemoryTransform

class torchrl.envs.transforms.PinMemoryTransform

在 tensordict 上呼叫 pin_memory 以便於在 CUDA 裝置上進行寫入操作。

forward(tensordict: TensorDictBase) → TensorDictBase

讀取輸入的 tensordict，並對選定的鍵應用變換。