Pooled self attention layer for 2d.
PooledSelfAttention2d(n_channels)
n_channels
number of channels
None