tensor-sensor

1
import numpy as np
2
import tsensor
3
import torch
4

5
W = np.array([[1, 2], [3, 4]])
6
b = np.array([9, 10]).reshape(2, 1)
7
x = np.array([4, 5]).reshape(2, 1)
8
h = np.array([1, 2])
9
# with tsensor.explain(savefig="/Users/parrt/Desktop/foo.pdf"):
10
#     W @ np.dot(b,b) + np.eye(2,2)@x
11

12

13
W = torch.tensor([[1, 2], [3, 4]], dtype=torch.float32)
14
b = torch.tensor([9, 10]).reshape(2, 1)
15
x = torch.tensor([4, 5], dtype=torch.int32).reshape(2, 1)
16
h = torch.tensor([1,2])
17

18
a = torch.rand(size=(2, 20), dtype=torch.float64)
19
b = torch.rand(size=(2, 20), dtype=torch.float32)
20
c = torch.rand(size=(2,20,200), dtype=torch.complex64)
21
d = torch.rand(size=(2,20,200,5), dtype=torch.float16)
22

23

24
with tsensor.explain(savefig="/Users/parrt/Desktop/t2.pdf"):
25
    a + b + x + c[:,:,0] + d[:,:,0,0]
26

27
with tsensor.explain(savefig="/Users/parrt/Desktop/t3.pdf"):
28
    c
29

30
with tsensor.explain(savefig="/Users/parrt/Desktop/t4.pdf"):
31
    d
32

33
# with tsensor.explain(legend=True, savefig="/Users/parrt/Desktop/t.pdf") as e:
34
#     W @ torch.dot(b, b) + torch.eye(2, 2) @ x
35