torch.take