torch.nn.utils.rnn.invert_permutation

torch.nn.utils.rnn.invert_permutation(permutation) [исходный код]

Возвращает обратную перестановку для permutation.

Это полезно для преобразования между отсортированными и неотсортированными индексами в PackedSequence.

Параметры:

permutation (Tensor, необязательно) – одномерный тензор индексов для инвертирования

Тип возвращаемого значения:

Tensor | None