As they are tensors, convert the type to int so as to index into the tensor of images and retrieve the respective patch for each batch. from torch. gather ...
確定! 回上一頁