食品网站建设网站定制开发,百度识图搜索引擎,微信官网网页版登录入口,seo实战密码电子版1.作用
从输入张量中按照指定维度进行索引采集操作#xff0c;返回值是一个新的张量#xff0c;形状与 index 张量相同#xff0c;根据指定的索引从输入张量中采集对应的元素。
2.问题
该函数的主要问题主要在dim维度上#xff0c;dim0 表示沿着第一个维度#xff08;行…1.作用
从输入张量中按照指定维度进行索引采集操作返回值是一个新的张量形状与 index 张量相同根据指定的索引从输入张量中采集对应的元素。
2.问题
该函数的主要问题主要在dim维度上dim0 表示沿着第一个维度行进行索引采集而 dim1 表示沿着第二个维度列进行索引采集。 简单讲dim0将在行上进行采集行数不变在列上取值如下图中的例子torch.gather(input, dim0, indexindex),当dim0时[0,1]中0对应第一行第一列也就是1,1对应着第2行第2列1在index的坐标为1,2dim0所以不用看index的行坐标的只管纵坐标也就是第2列。而此时值为1代表值input的行【需要1下标是从0开始的】也就是第2行值为4也就是4[1,0]中的1代表第2行第1列也就是3,0也就是第2列的第一行数据此时index的0的坐标为2,2因为dim0也就不用看横坐标也就是第二列。所以此时的0代表源input的行坐标也就是第一行也就是2
input torch.tensor([[1, 2], [3, 4], [5, 6]])
index torch.tensor([[0, 1], [1, 0]])result torch.gather(input, dim0, indexindex)
result_colum torch.gather(input, dim1, indexindex)
print(result:,result)
print(result_colum:,result_colum)结果如下
