Self attention layer for `n_channels`.

SelfAttention(n_channels)

Arguments

n_channels

number of channels

Value

None