deep learning激活函數實現之cudnn

代碼不方便上傳，只能傳圖片。 注意：cudnn對於輸入的維度限制爲4~8，少於4維的數據可以把多餘的維度的shape設置爲1，多餘8維的數據則無能爲力了。cudnn對於輸入數據的size限制爲最多20億，也就是說輸入數組的元素的個數最多爲20億。 先看kernel頭文件： kernel cpp文件： run:

>>阅读原文<<