快捷方式

torchrl.trainers.algorithms.configs.transforms.BatchSizeTransformConfig

class torchrl.trainers.algorithms.configs.transforms.BatchSizeTransformConfig(batch_size: list[int] | None = None, reshape_fn: Any = None, reset_func: Any = None, env_kwarg: bool = False, _target_: str = 'torchrl.envs.transforms.transforms.BatchSizeTransform')[源代码]

BatchSizeTransform 的配置。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源