torchtyping
Pythonpushed about 2 years ago
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
AI summary
Tensor checker
Provides type annotations and dynamic checking for tensor shapes and types in PyTorch
- stars
- 1.4K
- forks
- 34
- watching
- 16
- awesome list
- 1
Featured in 1 awesome list
Each link jumps to the spot where the list mentions torchtyping.