How to Build Your Own PyTorch Neural Network Layer from Scratch
And learn a thing or two about weight initialization
Published in
8 min readNov 1, 2019
This is actually an assignment from Jeremy Howard’s fast.ai course, lesson 5. I’ve showcased how easy it is to build a Convolutional Neural Networks from scratch using PyTorch. Today, let’s try to delve down even deeper and see if we could write our own nn.Linear
…