
人工神经网络
文章平均质量分 96
神经网络应用
欲买桂载酒
这个作者很懒,什么都没留下…
展开
-
使用Pytorch的 C++ (CUDA)扩展实现卷积算子
在实现卷积算子时,将卷积算子等价转换为两个阶段:收集(collecting)和 线性变换(linear transform)。在收集阶段,“image to column” 和 “column to image” 过程分别在前向传播中将输入的。文件 setup.py,在项目目录中使用命令 python setup.py install 将其编译安装到python中。这里根据设备不同提供两种实现方式,一种是在CPU设备上串行实现,另一种是在GPU设备上并行实现。在线性变换阶段,卷积核权重对前向传播中得到的。原创 2024-08-26 12:48:51 · 744 阅读 · 0 评论 -
Tensorflow2.6实现Unet结构神经网络(3D卷积)识别脑部肿瘤并实现模型并行
tensorflow2 实现Unet结构神经网络,并尝试实现模型并行,但是并没有达到预期效果。原创 2022-06-18 09:21:17 · 3063 阅读 · 4 评论 -
一个简单的神经网络Python实现(无框架)
记录一下自己在神经网络中学习结果并不使用框架实现一个含有卷积层和全连接层的神经网络。原创 2022-08-20 17:02:41 · 972 阅读 · 0 评论