import torch import numpy as np a=torch.linspace(1,25,steps=25).view(5,5) a[:,3:4]=0 b=np.asarray([[1,2]]) print(a) print(a[b])
import torch import numpy as np a=torch.linspace(1,25,steps=25).view(5,5) a[:,3:4]=0 b=np.asarray([[1,2]]) print(a) print(a[b])