You could use torch.randperm to create random indices and then index it using @asml's suggestion. 10 Likes.
確定! 回上一頁