| 1796 | // ------------------------------------------------------------------------------------ |
| 1797 | |
| 1798 | void softmax_all ( |
| 1799 | tensor& dest, |
| 1800 | const tensor& src |
| 1801 | ) |
| 1802 | { |
| 1803 | DLIB_CASSERT(have_same_dimensions(dest,src)); |
| 1804 | ttimpl::softmax(1, src.nr()*src.nc()*src.k(), dest, src); |
| 1805 | } |
| 1806 | |
| 1807 | void softmax_all_gradient ( |
| 1808 | tensor& grad, |