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,882
  • 7
  • 28
  • 45
Tob
  • 985
  • 1
  • 10
  • 26

0 Answers0