I want to get the product of the neuron outputs in the same layer, how to do effieciently?
for example, the last layer of my net has 16 neurons, hence the dimension of the output is (batch_size,16). I want to get the product of all 16 elements: (batch_size, product of all 16 elements ), how to achieve this ?
thanks