Update misc

2021-06-08 05:42:16 -07:00
parent 5d7ccd445d
commit f9bbf974de
3 changed files with 164 additions and 2 deletions
--- a/tests/test_super_rearrange.py
+++ b/tests/test_super_rearrange.py
@@ -26,3 +26,5 @@ class TestSuperReArrange(unittest.TestCase):
        tensor = torch.rand((8, 4, 32, 32))
        print("The tensor shape: {:}".format(tensor.shape))
        print(layer)
+        outs = layer(tensor)
+        print("The output tensor shape: {:}".format(outs.shape))