Solved by albanD in post #2 
              
              
                Hi, 
To do this, I usually use .narrow() and extract a single element. For example to get the second image: data.narrow(0, 1, 1)