| 13 | #include <cmath> |
| 14 | |
| 15 | inline void CalcSoftmax(const TConstArrayRef<double> approx, TArrayRef<double> softmax) { |
| 16 | double maxApprox = *MaxElement(approx.begin(), approx.end()); |
| 17 | for (size_t dim = 0; dim < approx.size(); ++dim) { |
| 18 | softmax[dim] = approx[dim] - maxApprox; |
| 19 | } |
| 20 | NCB::FastExpWithInfInplace(softmax.data(), softmax.ysize()); |
| 21 | double sumExpApprox = 0; |
| 22 | for (auto curSoftmax : softmax) { |
| 23 | sumExpApprox += curSoftmax; |
| 24 | } |
| 25 | for (auto& curSoftmax : softmax) { |
| 26 | curSoftmax /= sumExpApprox; |
| 27 | } |
| 28 | } |
| 29 | |
| 30 | inline TVector<double> CalcExponent(TVector<double> approx) { |
| 31 | NCB::FastExpWithInfInplace(approx.data(), approx.ysize()); |
no test coverage detected