TikTok interview question

Draw a neural network and write the loss function.