- 博客(3)
- 资源 (7)
- 问答 (3)
- 收藏
- 关注
原创 Pytorch学习 (二十一) ------自定义C++/ATen扩展
总说没办法, 出来混总是要还的, 不会写点底层代码没法混啊. 废话不多说, 简单来说, 有时候我们需要写一些自定义的操作, 这些操作如果用python写会很慢, 我们需要用CUDA写, 然后这些操作与python绑定, 以供python端调用.主要是简略拿出 https://pytorch.org/tutorials/advanced/cpp_extension.html 的东西, 根据实践,...
2019-03-27 22:42:28 8226 9
原创 Pytorch学习(二十) ------ 可能有用的代码合集
总说记录一些比较有用的pytorch代码(有些是自己写的, 有些是从网上看到的)目录提取网络特征(适用于sequential构建的网络)修改Pretrained的网络(如ResNet等)提取网络特征(适用于sequential构建的网络)class VGG16FeatureExtractor(nn.Module): def __init__(self): s...
2019-03-13 10:26:48 1844
为什么matlab中相同大小的矩阵运行速度差异巨大
2016-09-18
MATLAB的一个神经网络已经做好,想用C#写界面,如何才能调用?
2016-04-01
如何解决这个矩阵问题?
2016-03-23
TA创建的收藏夹 TA关注的收藏夹
TA关注的人