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.

by omriwen· Repository·documents
torch
shape
validator
Also installable via skills CLI
npx skills add omriwen/PRISM/.claude/skills/torch-shape-validator

Source

Repo:local
Path:.claude/skills/torch-shape-validator(main)

Related in documents