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·data
torch
shape
validator
Also installable via skills CLI
npx skills add omriwen/PRISM/data/torch-shape-validator

Source

Repo:local
Path:data/torch-shape-validator(main)

Related in data