PinMemoryTransform

class torchrl.envs.transforms.PinMemoryTransform

在 tensordict 上呼叫 pin_memory 以方便寫入 CUDA 裝置。

forward(tensordict: TensorDictBase) → TensorDictBase

讀取輸入的 tensordict，並針對選定的鍵套用轉換。