Pytorch
hawkcici160
好好学习吧
展开
-
torch.nn 和 torch.functional 的区别
原文地址 : https://blog.csdn.net/teeyohuang/article/details/79211798 Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)定义卷积神经网络注:官方文档地址– http://pytorch.org/docs/0...转载 2018-04-28 22:50:23 · 21155 阅读 · 1 评论 -
pytorch中的gather函数
from: https://blog.csdn.net/edogawachia/article/details/80515038今天刚开始接触,读了一下documentation,写一个一开始每太搞懂的函数gatherb = torch.Tensor([[1,2,3],[4,5,6]])print bindex_1 = torch.LongTensor([[0,1],[2,0]])i...转载 2018-06-22 11:12:27 · 3634 阅读 · 0 评论