
Python#函数小trick
诸葛灬孔暗
落日无边江不尽,此身此日更须忙
展开
-
【函数小trick】torch中scatter()、scatter_()详解(多标签one-hot向量生成)
1. 背景可以先看torch官方文档介绍:主要作用是根据索引值index,向tensor中指定dim维度的index位置写入scr所对应的数值,可以用来生成one-hot向量和特定mask,熟练使用该函数,就不用暴力for循环啦。2. 函数应用(1)one-hot向量生成将图像的标签值转换为一组只有0和1组成的向量,这是DL领域常用到的。import torchtargets=torch.zeros(3,5)index = torch.LongTensor([[3],[2]原创 2021-07-05 09:37:48 · 1427 阅读 · 1 评论 -
【函数小Trick】torch.gather(获取高维数据/矩阵/数组特定位置值)
1.背景可以先看torch官方文档介绍主要作用是根据索引值index,找出input向量中指定dim维度所对应的数值,熟练使用该函数,就不用暴力for循环啦。2.函数应用场景(1)自然语言处理中的mask与padding位置import torcha= torch.Tensor([ [4,1,2,0,0], [2,4,0,0,0], [1,1,1,6,5], [1,2,2,2,2], [3,0,0,0,0], [2,2,0,0,0]原创 2021-07-03 12:08:19 · 1443 阅读 · 1 评论