How do I copy, for example, the 1 index of the first dimension of a 3d tensor, and create a new tensor with   only that information?
For example in a tensor of [5,25,25], how do I copy [x,25,25] and create a new tensor [1,25,25]?