Solved by ptrblck in post #2 
              
              
                You can directly index tensors via tensor[] as e.g. in numpy: 
x = torch.tensor([[[0.3843, 0.3552]],
                [[0.1743, 0.2439]],
                [[0.3474, 0.1554]],
                [[0.1325, 0.0778]],
                [[0.2280, 0.1096]],
                [[0.3842, 0.3997]],
                [[0…