The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
import torch
import tsensor
n = 200 # number of instances
d = 764 # number of instance features
nhidden = 256
Whh = torch.eye(nhidden, nhidden) # Identity matrix
Uxh = torch.randn(nhidden, d)
bh = torch.zeros(nhidden, 1)
h = torch.randn(nhidden, 1) # fake previous hidden state h
r = torch.randn(nhidden, 3) # fake this computation
X = torch.rand(n,d) # fake input
with tsensor.explain(savefig):
r*h
There is some overlap with these boxes: