Back to problems

NumPy Conv2D Forward and Parameter Count

AI Coding · Tesla · Hard

Requirements From the input spatial dimensions, input/output channel counts, kernel dimensions, stride, and padding, determine the Conv2D output shape. Calculate the full parameter total, adding bias parameters whenever bias is turned on. Write the forward Conv2D computation with NumPy rather than using TensorFlow or PyTorch convolution primitives. Explain or build an improvement over straightforward nested loops that uses a more vectorized approach. Since one interview…

Checking your access…