# 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)$$