Squeeze-Excite Params

Problem

Count the parameters of the squeeze-and-excite bottleneck, excluding biases.

The first fully connected layer maps channels to channels // reduction and the second maps back. Return the total.

case 1

in: [64,16]

expected: 512

case 2

in: [128,8]

expected: 4096

case 3

in: [16,4]

expected: 128

Starter codePython

def se_params(channels, reduction):
    # Your code here
    pass

3 test cases · run your code against them in the editor