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.
https://github.com/parrt/tensor-sensor
MIT License
794 stars 40 forks source link

General clean up #30

Closed parrt closed 3 years ago

parrt commented 3 years ago

Mostly formatting, commenting, but some style stuff.

parrt commented 3 years ago

How does this look @sbrugman ?