You can use torch.take to get rid of a for loop. But first, an array of indices should be created with this function def convert_inds(img_a,img_b,patch_a ...
確定! 回上一頁