Solved by LeviViana in post #4 
              
              
                This works: 
>>> import torch
>>> a = torch.rand(2, 3, 4)
>>> print(a)
tensor([[[0.5066, 0.5184, 0.1193, 0.6062],
         [0.4995, 0.1689, 0.6175, 0.7917],
         [0.7996, 0.0225, 0.1145, 0.4249]],

        [[0.0975, 0.2995, 0.5857, 0.0806],
         [0.4922, 0.4778, 0.9133, 0.1418],
         [0.…