Как я могу заменить записи тензора с соответствующим рейтингом каждой записи? - PullRequest
Новая
       31

Как я могу заменить записи тензора с соответствующим рейтингом каждой записи?

0 голосов
/ 28 октября 2019

предположим, что у меня есть следующий тензор: 

>> i = 3
>> j = 5
>> k = 2
>> sor = torch.randn(i,j,k)

>> sor
Out[20]: 
tensor([[[ 0.5604, -0.9675],
         [-1.0953, -0.5615],
         [ 0.4250, -0.9176],
         [-1.6188, -1.0217],
         [-0.0778,  1.9407]],

        [[-0.1034, -0.7925],
         [-0.2955,  0.8058],
         [-0.5349,  1.1040],
         [ 1.1240,  0.8249],
         [ 0.0827, -1.2471]],

        [[ 0.5924,  0.4777],
         [-2.4640, -1.9527],
         [-0.4519,  0.4788],
         [-0.2308, -0.2368],
         [-1.6786,  0.1360]]])

предположим, что для каждых фиксированных i и j я хочу вычислить числовой ранг элементов для k и заменитьэлементы тензора sor с этими рангами. Например, из приведенного выше примера я хочу изменить запись [ 0.5604, -0.9675], которая равна sor[0,0,:], на [1, 2], поскольку 0.5604 > -0.9675

Спасибо,

1 Ответ

2 голосов
/ 28 октября 2019

Я думаю, что вы ищете torch.argsort: 

torch.argsort(sor, dim=2)

Out[ ]:
tensor([[[1, 0],
         [0, 1],
         [1, 0],
         [0, 1],
         [0, 1]],

        [[1, 0],
         [0, 1],
         [0, 1],
         [1, 0],
         [1, 0]],

        [[1, 0],
         [0, 1],
         [0, 1],
         [1, 0],
         [0, 1]]])
...