parrt/tensor-sensor
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
When building deep learning models, it's common for tensor (matrix/vector) dimensions to get mismatched, leading to confusing error messages. This tool helps machine learning engineers and data scientists quickly identify and fix these dimension errors. You input your Python code that performs tensor operations, and it outputs clearer error messages and visualizations showing the exact shapes of your tensors at each step.
810 stars. No commits in the last 6 months. Available on PyPI.
Use this if you are a machine learning engineer or data scientist struggling to debug complex tensor dimension errors in frameworks like TensorFlow, PyTorch, JAX, or NumPy.
Not ideal if your tensor operations are spread across multiple lines within function definitions or use backslash continuations, as these are not currently supported.
Stars
810
Forks
39
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Apr 07, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/parrt/tensor-sensor"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
pymc-devs/pytensor
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions...
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
lava-nc/lava-dl
Deep Learning library for Lava
tensorly/tensorly
TensorLy: Tensor Learning in Python.
tensorpack/tensorpack
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility