Uploaded image for project: 'SystemML'
  1. SystemML
  2. SYSTEMML-1676

Add a new 2D softmax layer

    XMLWordPrintableJSON

    Details

    • Type: New Feature
    • Status: Closed
    • Priority: Major
    • Resolution: Fixed
    • Affects Version/s: None
    • Fix Version/s: SystemML 0.15
    • Component/s: Algorithms
    • Labels:
      None

      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.

        Attachments

          Activity

            People

            • Assignee:
              Tenma Fei Hu
              Reporter:
              dusenberrymw Mike Dusenberry
            • Votes:
              0 Vote for this issue
              Watchers:
              1 Start watching this issue

              Dates

              • Created:
                Updated:
                Resolved: