Caffe：math_function.hpp

本文主要是介绍Caffe：math_function.hpp，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

caffe_set函数定义在math_function.hpp头文件

template <typename Dtype>
void caffe_set(const int N, const Dtype alpha, Dtype *X);

template <typename Dtype>
void caffe_set(const int N, const Dtype alpha, Dtype* Y) {if (alpha == 0) {memset(Y, 0, sizeof(Dtype) * N);  // NOLINT(caffe/alt_fn)return;}for (int i = 0; i < N; ++i) {Y[i] = alpha;}
}

 

这篇关于Caffe：math_function.hpp的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---