Line data Source code
1 : #include "Softmax.h" 2 : 3 : namespace elsa::ml 4 : { 5 : template <typename data_t> 6 0 : Softmax<data_t>::Softmax(index_t axis, const std::string& name) 7 0 : : Layer<data_t>(LayerType::Softmax, name), axis_(axis) 8 : { 9 0 : } 10 : 11 : template <typename data_t> 12 0 : index_t Softmax<data_t>::getAxis() const 13 : { 14 0 : return axis_; 15 : } 16 : 17 : template <typename data_t> 18 0 : void Softmax<data_t>::computeOutputDescriptor() 19 : { 20 0 : this->outputDescriptor_ = this->inputDescriptors_.front()->clone(); 21 0 : } 22 : 23 : template class Softmax<float>; 24 : } // namespace elsa::ml