Drop output from a Neural Network

Question

I have a simple net:

net = NetInitialize@
  NetChain[{LinearLayer[2], SoftmaxLayer[]}, "Input" -> 2, 
   "Output" -> NetDecoder[{"Class", {0, 1}}]]

net@RandomReal[{-1, 1}, {10, 2}]

How do I return probabilites after SoftmaxLayer? Right now I have to insert dummy ElementwiseLayer[Identity] in the end of a chain and then Take[net,2] would return what I actually need.

Answer 1

maybe you can try this:

num=RandomReal[{-1, 1}, {10, 2}];
net[num,"Probabilities"]

And there is a new function NetDrop in 11.3.0, you can check:

SoftmaxLayer["Input" -> 2]@NetDrop[net, -1][num]

Exactly the same.