ELK>> Neural-Network>> 返回
项目作者： mbs0221

项目描述 ：
C++ Implementation of Neural-Network Framework
高级语言： C++
项目地址: git://github.com/mbs0221/Neural-Network.git
创建时间： 2019-10-22T08:05:59Z
项目社区：https://github.com/mbs0221/Neural-Network
开源协议：GNU General Public License v2.0
下载


Neural-Network

C++神经网络框架

  • 初步实现了反向传播算法，支持搭建stacked-model
  • 支持5D张量(sample, frame, width, height, channel)
  • 已实现如下基本运算类型
    • matmul/add
    • element-wise operation
    • broadcast operation
    • conv2d/conv3d operation
    • max/min/average pooling
    • max/min/average upsampling
    • permute/reshape/flatten
    • reduce_sum/reduce_mean
    • sigmoid/relu/leaky_relu/softmax
  • 初步实现了计算图框架，支持多分支网络模型
  • 预计进一步实现：
    • 基于梯度的一些优化算法