peal.architectures.basic_modules.Unsqueeze

class peal.architectures.basic_modules.Unsqueeze[source]

Bases: Module

Insert singleton dimensions, one torch.unsqueeze per entry of dims.

Parameters:

dims (list of int) – Dimensions to insert, applied in order (later entries see the already expanded shape).

Methods

forward

Unsqueeze x at every configured dimension.
__init__(dims)[source]
forward(x)[source]

Unsqueeze x at every configured dimension.