快捷方式

随机策略

torchrl.envs.utils.RandomPolicy(action_spec: TensorSpec, action_key: NestedKey = 'action')[source]

用于数据收集器的随机策略。

这是对 action_spec.rand 方法的包装器。

参数::

action_spec – 描述动作规范的 TensorSpec 对象

示例

>>> from tensordict import TensorDict
>>> from torchrl.data.tensor_specs import BoundedTensorSpec
>>> action_spec = BoundedTensorSpec(-torch.ones(3), torch.ones(3))
>>> actor = RandomPolicy(action_spec=action_spec)
>>> td = actor(TensorDict({}, batch_size=[])) # selects a random action in the cube [-1; 1]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源