Compute Conv2D parameter counts
Company: Tesla
Role: Machine Learning Engineer
Category: Machine Learning
Difficulty: easy
Interview Round: Technical Screen
For a 2D convolution layer, given input channels C_in, output channels C_out, kernel size (k_h, k_w), stride (s_h, s_w), padding (p_h, p_w), and an optional bias term, compute the total number of learnable parameters for the layer in both cases: with bias and without bias.
Quick Answer: This question evaluates understanding of convolutional neural network parameterization, specifically how kernel dimensions, input/output channels and an optional bias term determine the number of learnable parameters.