[
https://issues.apache.org/jira/browse/SYSTEMML-1676?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Mike Dusenberry updated SYSTEMML-1676:
--------------------------------------
Component/s: Algorithms
> Add a new 2D softmax layer
> --------------------------
>
> Key: SYSTEMML-1676
> URL: https://issues.apache.org/jira/browse/SYSTEMML-1676
> Project: SystemML
> Issue Type: New Feature
> Components: Algorithms
> Reporter: Mike Dusenberry
> Assignee: Fei Hu
> Fix For: SystemML 1.0
>
>
> A 2D softmax layer would accept a tensor of shape {{(N,C,H,W)}}, where the
> {{C}} axis contains scores for {{D}} classes, and output a tensor of the same
> shape, with the scores transformed to normalized probabilities. The typical
> use case would be a segmentation problem, in which every pixel has a
> multiclass prediction.
--
This message was sent by Atlassian JIRA
(v6.4.14#64029)