Type of layer

Activation

Output dimensions

One hot input

None

L × 2k

Conv1D

Relu

L × 32

RSBU

Relu

L × 32

Conv1D

Relu

L × 64

RSBU

Relu

L × 64

Conv1D

Relu

L × 256

RSBU

Relu

L × 256

Power Norm Layer

None

L × 256

AWGN channel

None

L × 256

Conv1D

Relu

L × 128

RSBU

Relu

L × 128

Conv1D

Relu

L × 64

RSBU

Relu

L × 64

Conv1D

Relu

L × 32

RSBU

Relu

L × 32

Conv1D

Relu

L × 2k

RSBU

Relu

L × 2k

Conv1D

Softmax

L × 2k