pytorch 学习
weich37
这个作者很懒,什么都没留下…
展开
-
Pytorch中tensor的索引
Pytorch中tensor的索引引言分类概述例子1、0维索引2、1维索引3、2维索引参考引言做了一段时间的目标检测,在这个过程中也复现了不少经典的检测网络,例如,faster-rcnn,yolov3,retinanet等等。在学习,消化别人代码的过程中,经常会遇到一个多维的tensor来索引另一个多维的tensor这种种类似的情况,而我对索引的概念还停留在a[0],a[0,:],a[0,…]...原创 2019-11-29 14:22:21 · 3532 阅读 · 1 评论 -
torch.nn.Embedding理解
看官方的文档之后,自己的理解,供以后学习之用。输入的batch为2,每个batch有4个索引input = torch.tensor([[1,2,4,5],[4,3,2,9]])字典中包含的词有10个,每个3维embedding_matrix = torch.rand(10, 3)F.embedding(input, embedding_matrix)tensor([[[ 0.8...原创 2019-02-27 14:39:11 · 5190 阅读 · 4 评论