ARTICLE DETAIL

资讯详情

深耕网站建设、视觉设计与SEO优化的一线实战洞察。

Pytorch随笔:gather函数

Pytorch随笔:gather函数

gather函数使用方法:
gather(dim,index)

dim = 0时,使用index填充行;dim = 1时,使用index填充列。
例如,index = tensor([[x,y,z]])
gather(dim=0,index)示意图为:
tensor([[(x,0),(y,1),(z,2)]])

gather(dim=1,index)示意图为:
tensor([[(0,x),(1,y),(2,z)]])

示例:
image
image

返回列表