网站集群建设,成都健康码小程序,免费设计logo在线生成,佛山市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)结果如下
