Apply to Softmax row
7 次查看(过去 30 天)
显示 更早的评论
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
1 个评论
采纳的回答
Mahesh Taparia
2021-5-31
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!
0 个评论
更多回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!