| 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 © 2016. All Rights Reserved.