fibertree-project / fibertree Goto Github PK

z = Tensor(rank_ids=["M", "N"])
z.setName("Z")
z.setMutable(True)

z_m = z.getRoot()
#
# Hack to fill in all the entries in z
# This allows us to pretend the tensor is dense
#
n_fiber = Fiber(coords=range(N), initial=1)
m_fiber = Fiber(coords=range(M), initial=1)

for m, (z_n, _) in z_m << m_fiber:
    for n, (z_ref, _) in z_n << n_fiber:
        z_ref <<= 0
        
z.getRoot()[0][0]

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[11], line 17
     14     for n, (z_ref, _) in z_n << n_fiber:
     15         z_ref <<= 0
---> 17 z.getRoot()[0][0]

File /usr/local/lib/python3.8/dist-packages/fibertree/core/fiber.py:1874, in Fiber.__getitem__(self, keys)
   1871     key += len(self)
   1873 if key < 0 or key >= len(self):
-> 1874     raise(IndexError, f"The index ({key}) is out of range")
   1876 new_payload = self.payloads[key]
   1878 if len(key_cdr):
   1879     # Recurse down the fiber tree

TypeError: exceptions must derive from BaseException