图解pytorch里面的torch.gather()_资讯

创始人

2024-03-25 04:40:02

0次

上图显示了 torch gather() 函数在 dim=1 的二维张量上的工作。

这里索引张量的行对应于输入张量的行（用灰色阴影突出显示）。现在对于索引张量中的每个索引值，从该行和输入张量的索引中选取相应的值。

让我们一步步理解这个例子——

代码实现如下所示—— 在 [5] 中：

torch.gather(input=tensor2, dim=1, index = torch.tensor([[2, 1, 0], [0, 2, 1]]))

出[5]：

张量（[[2, 1, 0],[3, 5, 4]])

参考文献: [Diagram] How to use torch.gather() Function in PyTorch with Examples - MLK - Machine Learning Knowledge

用心做好每一块电池的欣旺达，因... 这两天国内动力电池生产厂商欣旺达遇到麻烦事了，因其所生产的电芯存在质量问题被威睿电动汽车技术（宁波）...

以案为鉴筑防线以审促廉扬清风... 为充分发挥以案释法、以案说纪的警示教育作用，进一步加强党风廉政建设，提高党员干部的法纪意识和廉洁意识...

新加坡国立大学东亚研究所高级研... 由三亚市人民政府主办，《财经》杂志、财经网、《财经智库》、三亚中央商务区管理局、三亚经济研究院承办的...

原创全... 在国家有关调查力量进驻南京之后，一个并不显眼、却耐人寻味的现象悄然出现了。短时间内，全国多地博物馆...

跨境金融研究院院长王志毅：离岸... 由三亚市人民政府主办，《财经》杂志、财经网、《财经智库》、三亚中央商务区管理局、三亚经济研究院承办的...

原告向法官出示证据，右下角赫然... 近日，湖北孝感大悟法院民二庭在审理一起房屋租赁合同纠纷案时，精准识破原告方利用AI技术伪造证据的行为...

美国纽约州出台法律约束“成瘾性... 美国纽约州州长凯茜·霍楚尔26日宣布，根据该州新出台的一项法律，具备无限刷新、自动播放和算法推送功能...

富安娜理财纠纷一审落槌，中信证... 乐居财经李兰经历近三年后，富安娜(002327.SZ)理财纠纷有了新进展。 12月25日，富安娜发...

从合作伙伴到对簿公堂：威睿起诉... 12月26日，欣旺达发布公告，其全资子公司欣旺达动力科技股份有限公司（下称“欣旺达动力”）因买卖合同...

突发！俄称已控制库皮扬斯克；泽... 俄乌，突传大消息！俄国防部称已控制库皮扬斯克俄罗斯国防部12月27日在每日例行通报中说，库皮扬斯...