0

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]?

Md. Rezwanul Haque
  • 2,802
  • 5
  • 26
  • 41
Tob
  • 791
  • 1
  • 9
  • 21

0 Answers0