PyTorch 笔记(06)— Tensor 索引操作(index_select、masked_select、non_zero、gather)
Tensor支持与numpy.ndarray类似的索引操作,如无特殊说明,索引出来的结果与源tensor共享内存,即修改一个,另外一个也会跟着改变。In[65]:a=t.arange(0,6).reshape(2,3)In[66]:aOut[66]:tensor([[0,1,2],[3,4,5]])1.初级索引1.获取第0行In[67]:a[0]#第0行Out[67]:tensor([0,1,2]