评价此页

torch.nn.utils.rnn.invert_permutation#

torch.nn.utils.rnn.invert_permutation(permutation)[源码]#

返回 permutation 的逆置换。

这对于在 PackedSequence 中转换已排序和未排序的索引非常有用。

参数

permutation (Tensor, 可选) – 用于反转的索引的一维张量

返回类型

Optional[Tensor]