快捷方式

torchrl.trainers.algorithms.configs.transforms.DeviceCastTransformConfig

class torchrl.trainers.algorithms.configs.transforms.DeviceCastTransformConfig(device: str = 'cpu', in_keys: list[str] | None = None, out_keys: list[str] | None = None, in_keys_inv: list[str] | None = None, out_keys_inv: list[str] | None = None, _target_: str = 'torchrl.envs.transforms.transforms.DeviceCastTransform')[原始碼]

DeviceCastTransform 的配置。

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源