[
https://issues.apache.org/jira/browse/SYSTEMML-1676?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Mike Dusenberry updated SYSTEMML-1676:
--------------------------------------
Description: 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.
> Add a new 2D softmax layer
> --------------------------
>
> Key: SYSTEMML-1676
> URL: https://issues.apache.org/jira/browse/SYSTEMML-1676
> Project: SystemML
> Issue Type: Sub-task
> Reporter: Mike Dusenberry
> Assignee: Mike Dusenberry
>
> 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)
