UnsqueezeTransform

class torchrl.envs.transforms.UnsqueezeTransform(*args, **kwargs)

在指定位置插入大小为 1 的维度。

参数:

• unsqueeze_dim (int) – 要压缩的维度。必须为负数（或 allow_positive_dim 必须打开）。

• allow_positive_dim (bool, optional) – 如果为 True，则接受正维度。 UnsqueezeTransform 将这些映射到输入张量的第 n 个特征维度（即批处理大小后的第 n 个维度），独立于 tensordict 批处理大小（即正维度在传递不同批处理大小的 tensordict 的上下文中可能是危险的）。默认为 False，即不允许非负维度。

transform_input_spec(input_spec)

转换输入规范，以使结果规范匹配转换映射。

参数:

input_spec (TensorSpec) – 转换前的规范

返回:

转换后的预期规范

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

转换观察规范，以使结果规范匹配转换映射。

参数:

observation_spec (TensorSpec) – 转换前的规范

返回:

转换后的预期规范

transform_reward_spec(reward_spec: TensorSpec) → TensorSpec

转换奖励规范，以使结果规范匹配转换映射。

参数:

reward_spec (TensorSpec) – 转换前的规范

返回:

转换后的预期规范