[
https://issues.apache.org/jira/browse/SYSTEMML-1676?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16070867#comment-16070867
]
Mike Dusenberry commented on SYSTEMML-1676:
-------------------------------------------
cc [~niketanpansare] This can be used in Caffe2DML now.
> Add a new 2D softmax layer
> --------------------------
>
> Key: SYSTEMML-1676
> URL: https://issues.apache.org/jira/browse/SYSTEMML-1676
> Project: SystemML
> Issue Type: New Feature
> 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)
