[ https://issues.apache.org/jira/browse/SYSTEMML-1676?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Glenn Weidner updated SYSTEMML-1676: ------------------------------------ Fix Version/s: (was: SystemML 1.0) SystemML 0.15 > 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 0.15 > > > 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)