torch-shape-validator
Add tensor shape validation and documentation to PyTorch code. This skill should be used when working with PyTorch models to ensure tensor shapes are correct and well-documented with inline comments.
torch
shape
validator
Also installable via skills CLI
npx skills add omriwen/PRISM/.claude/skills/torch-shape-validator