过滤器大小包含N * kernel_size * kernel_size
权重参数,每个通道一个,所以
N * kernel_size * kernel_size* n_channels
然后N bais parmaters,所以该层的最终cacluation为n_params = N * kernel_size * kernel_size* n_channels + N
N : number of features
kernel_size : is conv2d shape (height and width)
n_channels : is the number of channels
n_params : is the total number of your parameters
Ex
n_params = 64 * (4 * 4) * 3 + 64 = 3136