I have a problem.
X is a Variable with data type torch.cuda.FloatTensor of size [batch, channel, height, width], I need to index X with another array, e.g., Y = X[[0,0,1,1], [0,0,1,2]]. Just like the indexing function implemented in numpy. Here is the link: https://docs.scipy.org/doc/numpy-1.13.0/reference/arrays.indexing.html 5
However, the following cuda runtime error occurs:
RuntimeError: cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1513368888240/work/torch/lib/THC/generic/THCTensorCopy.c:u6708:
Can anybody help me? Thank you!