SqueezeLayer

class torchrl.modules.SqueezeLayer(dims: Sequence[int] = (- 1,))

挤压层。

挤压输入张量的一些给定单例维度。

参数:

dims (iterable) – 要挤压的维度，默认值：(-1,)

forward(input: Tensor) → Tensor

定义每次调用时执行的计算。

应被所有子类重写。

注意

虽然正向传递的配方需要在该函数内定义，但之后应该调用 Module 实例而不是此函数，因为前者负责运行注册的钩子，而后者会默默地忽略它们。