Pytorch, Torch, Torchvision, Deep Learning, Python.
0.4351, 0.8555, 0.1771, 0.3130], [0.0801, 0.3683, 0.7401, 0.6129, 0.3549]]) indices = torch.LongTensor([0, 5, 4]) # Split on the rows split = torch.index_select(x, 0, indices) print(split) >> tensor([[0.0602, 0.8056, 0.9265, 0.3025, 0.6211], [0.0990, 0.0997, 0.9118, 0.1010, 0.8384], [0.8034,