Derive MLP shapes and explain PyTorch broadcasting

Company: NVIDIA

Role: Software Engineer

Category: Machine Learning

Difficulty: medium

Interview Round: Technical Screen

Quick Answer: This question evaluates understanding of tensor shapes, linear layer forward computation, and deep-learning-framework broadcasting semantics. It is commonly asked to confirm practical ability to translate mathematical layer definitions into concrete tensor shapes and to reason about broadcasting behavior; it belongs to the Machine Learning domain and tests practical application of tensor algebra grounded in conceptual understanding.