The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
copied from cf-staging / tensor-sensorTensorSensor clarifies exceptions by augmenting messages and visualizing Python code to indicate the shape of tensor variables.
