-
Type:
New Feature
-
Status: Closed
-
Priority:
Major
-
Resolution: Fixed
-
Affects Version/s: None
-
Fix Version/s: SystemML 0.15
-
Component/s: Algorithms
-
Labels:None
-
Epic Link:
-
Sprint:Sprint 1
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.