Let's say the second to last layer of my neural network outputs a tensor with shape [7].
How would I make my output layer be only a subset of the previous 7 outputs, for example an array of the values at index 0, 3, and 6? 
For example, if my second to last output is [0.2, 1, 6, 1, 2, 0.87, 1.7], how could I get the last layer to output [0.2, 1, 1.7]?
 
    