numpy pytorch索引

numpy python 索引实在是精妙，本文记录遇到的索引逻辑比较好的代码和问题

:代表切(不会变维度,只会变大小[5,6]->[2,4])，数字代表选，[]:相当于多选，比选多一维罢了

import torch
x=torch.arange(120).reshape(2,3,4,5)
print(x)
print(x[:2,])
print(x[[0,1],[1,1],[2,2]])
print(x[0,1,2])
print(x[1,1,2])

import torch
import math
data=torch.arange(120).reshape(2,3,4,5)
print(data)
print(data[:,1,:,1:3])

import torch
import math
data=torch.arange(120).reshape(2,3,4,5)
print(data)
index=data[:,:,:,0]
print(index)
mask=(index>50)
print(mask)
print(data[mask])

import torch
weight=torch.rand(3, 4)

index=input_indices = torch.randint(0, 3, (2,3))
print(weight)
print(index)
print(weight[index])