FlattenTime#

class sinabs.layers.FlattenTime[source]#

Utility layer which always flattens first two dimensions and is a special case of torch.nn.Flatten(). Meant to convert a tensor of dimensions (Batch, Time, Channels, Height, Width) into a tensor of (Batch*Time, Channels, Height, Width).

Shape:
  • Input: \((Batch, Time, Channel, Height, Width)\) or \((Batch, Time, Channel)\)

  • Output: \((Batch \times Time, Channel, Height, Width)\) or \((Batch \times Time, Channel)\)