Skip to content Skip to sidebar Skip to footer
Showing posts with the label Numpy Slicing

Index Multidimensional Torch Tensor By Another Multidimensional Tensor

I have a tensor x in pytorch let's say of shape (5,3,2,6) and another tensor idx of shape (5,3,… Read more Index Multidimensional Torch Tensor By Another Multidimensional Tensor