espnet2.enh.layers.swin_transformer.window_reverse

source

espnet2.enh.layers.swin_transformer.window_reverse(windows, window_size, H, W)

Restore the windowed feature into its original shape.

• Parameters:
  • windows – (num_windows*B, window_size, window_size, C)
  • window_size (Tuple *[*int , int ]) – Window size
  • H (int) – Height of the 2D feature
  • W (int) – Width of the 2D feature
• Returns: (B, H, W, C)
• Return type: x