Apply to Softmax row

As far as I know, the Softmax function is applied per channel. How do I apply this per row? And how do I apply this to the custom layer

[KALYAN ACHARJYA]

If you tranpose the data (rotate), then apply on resultant.

Answer 1

[Mahesh Taparia]

Hi

It seems you want to perform the softmax function as per your data/feature dimension. In this case you can create a custom loss layer and define the softmax operation inside that as per your need. You can refer this documentation for custom loss function creation. Also, refer this documentation for custom training loop workflow. Moreover, refer this documentation for custom layer creation.

Hope it will help!