Вычисление попарных расстояний между записями в `torch.tensor` ⇐ Python

[Anonymous]

Я пытаюсь реализовать тип потери выравнивания многообразия, показанный здесь.
Например, с учетом тензора, представляющего пакет вложений формы (L,N) с L=256:

Код: Выделить всё

tensor([[ 0.0178,  0.0004, -0.0217,  ..., -0.0724,  0.0698, -0.0180],
[ 0.0160,  0.0002, -0.0217,  ..., -0.0725,  0.0655, -0.0207],
[ 0.0155, -0.0010, -0.0153,  ..., -0.0750,  0.0688, -0.0253],
...,
[ 0.0130, -0.0113, -0.0078,  ..., -0.0805,  0.0634, -0.0241],
[ 0.0120, -0.0047, -0.0135,  ..., -0.0846,  0.0722, -0.0230],
[ 0.0120, -0.0048, -0.0142,  ..., -0.0843,  0.0734, -0.0246]],
grad_fn=)

Я хочу вычислить все попарные расстояния между записями строк. В результате получается результат в форме (L, L).
Я пробовал использовать torch.nn.PairwiseDistance, но мне не ясно, так ли это полезно для того, что я ищу.

Подробнее здесь: https://stackoverflow.com/questions/753 ... rch-tensor