Visualise PyTorch Tensors as Lego blocks
Been working on a visualisation engine for PyTorch tensors.
When I was learning PyTorch, tensors really started to click once I stopped thinking of them as arrays and started thinking of them more like Lego blocks — things you can slice, reshape, stack, repeat, squeeze, and combine.
So I built a visualisation library to make those operations tangible.
Write a PyTorch operation and actually see what it does to the tensor.
A huge amount of machine learning ultimately comes down to manipulating tensors. Once you can build an intuition for their shape and how operations transform them, a lot of PyTorch starts to feel much less abstract.
Would love to hear whether something like this would have helped when you were learning PyTorch and if you'd like me to open source this :)