How to retain 2D (or more) shape when using pytrorch masked_select

后端未结

关注

 0  1769

Suppose I have the following two matching shape tensors:

a = tensor([[ 0.0113, -0.1666,  0.5960, -0.0667], [-0.0977, -0.1984,  0.5153,  0.0420]])
selectors = tens


                      
              相关标签: