PooledSelfAttention2d

Source: R/layers.R

Pooled self attention layer for 2d.

PooledSelfAttention2d(n_channels)

Arguments

n_channels

number of channels

Value

None