Package | Description |
---|---|
org.nd4j.linalg.api.ops.impl.transforms |
Class and Description |
---|
SoftMax
Soft max function
row_maxes is a row vector (max for each row)
row_maxes = rowmaxes(input)
diff = exp(input - max) / diff.rowSums()
Outputs a probability distribution.
|
Copyright © 2015. All Rights Reserved.