1、主要有两个参数:dim即维度指示,index即该维度的具体索引;
2、输出结果的shape和index完全一致;
例子:
图中示例1,index是个[2, 2]的矩阵,那么结果也是一个[2, 2]的矩阵,如下图:
dim=0时,index即行索引,把索引先填上去:
(0, ) | (0,) |
(1,) | (1,) |
这些索引是对源矩阵的定位,所以还需要列索引,列索引直接按照表格自然位置填充即可:
(0,0 ) | (0,1) |
(1,0) | (1,1) |
最后一步,取数,例如(0,0)表示获取源矩阵0行0列,即源矩阵左上角的1,以此类推。