Update delta-iris/src/tokenizer.py

delta-iris/src/tokenizer.py

@@ -75,7 +75,7 @@ class Tokenizer(nn.Module):
         e = kle // (k * l)
 
         q = q.reshape(b, t, h, w, k, l, e)
-        q = q.transpose(0, 1, 6, 2, 4, 3, 5)
+        q = q.permute(0, 1, 6, 2, 4, 3, 5)
         q = q.reshape(b, t, e, h * k, w * l)
 
         return q