CUDNN學習筆記(1)

cuDNN概述 NVIDIA cuDNN是一個GPU加速深層神經網絡原語庫。它提供了在DNN應用程序中頻繁出現的例程的高度優化的實現: 卷積前饋和反饋, pooling前饋和反饋 softmax前饋和反饋 神經元前饋和反饋: 整流線性(ReLU) -sigmoid 雙曲線正切(TANH) 張量轉換函數 LRN,LCN和批量歸一化前進和後退 cuDNN的卷積程序旨在提高性能,以最快的GEMM(矩陣乘
相關文章
相關標籤/搜索