peal.architectures.basic_modules.Squeeze

class peal.architectures.basic_modules.Squeeze[source]

Bases: Module

Remove singleton dimensions, one torch.squeeze per entry of dims.

Parameters:

dims (list of int) – Dimensions to squeeze, applied in order.

Methods

forward

Squeeze x at every configured dimension.

__init__(dims)[source]
forward(x)[source]

Squeeze x at every configured dimension.