tweaks; remove a field and make it a temporary variable. the three-dimensional text should be on the right side of the box overdrawn on the edges; shifting it back.

Author: parrt

testing/test2.py

@@ -8,7 +8,7 @@
 h = torch.tensor([1,2])
 
 with tsensor.explain(legend=True, savefig="/Users/parrt/Desktop/t2.pdf"):
-    W @ torch.eye(2, 2)
+    torch.rand(size=(2,20,2000,10))
 
-with tsensor.explain(legend=True, savefig="/Users/parrt/Desktop/t.pdf") as e:
-    W @ torch.dot(b, b) + torch.eye(2, 2) @ x
+# with tsensor.explain(legend=True, savefig="/Users/parrt/Desktop/t.pdf") as e:
+#     W @ torch.dot(b, b) + torch.eye(2, 2) @ x

tsensor/viz.py

@@ -150,9 +150,9 @@ def set_locations(self, maxh):
         That is why this is a separate function not part of the constructor.
         """
         line2text = self.hchar / 1.7
-        self.box2line = line2text*2.6
-        self.texty = self.bottomedge + maxh + self.box2line + line2text
-        self.liney = self.bottomedge + maxh + self.box2line
+        box2line = line2text*2.6
+        self.texty = self.bottomedge + maxh + box2line + line2text
+        self.liney = self.bottomedge + maxh + box2line
         self.box_topy = self.bottomedge + maxh
         self.maxy = self.texty + 1.4 * self.fontsize
 
@@ -294,7 +294,6 @@ def draw_matrix(self,ax,sub, sh, ty):
                 fontname=self.dimfontname, fontsize=self.dimfontsize, rotation=90)
 
         # Note: this was always true since matrix...
-        # if len(sh) > 1:
         textx = mid
         texty = self.box_topy + self.dim_ypadding
         if len(sh) > 2:
@@ -308,7 +307,7 @@ def draw_matrix(self,ax,sub, sh, ty):
         if len(sh) > 2:
             # Text to the right
             ax.text(box_left+w, self.box_topy - h/2, self.nabbrev(sh[2]),
-                    verticalalignment='center', horizontalalignment='right',
+                    verticalalignment='center', horizontalalignment='center',
                     fontname=self.dimfontname, fontsize=self.dimfontsize,
                     rotation=45)