I made a free VS Code extension that shows tensor shapes while you step through your model
When I was learning, half my bugs were shape mismatches, and I kept typing x.shape into the debug console. dims shows the shape of every tensor/array right in the debugger's Variables view: x = {[32, 784]}, tensor(...) logits = {[32, 10]}, tensor(...) df = {[10000, 12]}, DataFrame So you can step l…
Read the full story at r/learnmachinelearning ↗
Timeline · 1 report
- 2026-10-23 13:56 · r/learnmachinelearning
I made a free VS Code extension that shows tensor shapes while you step through your model