當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.math.invert_permutation用法及代碼示例


計算張量的逆排列。

用法

tf.math.invert_permutation(
    x, name=None
)

參數

  • x 一個Tensor。必須是以下類型之一:int32 , int64。一維。
  • name 操作的名稱(可選)。

返回

  • 一個Tensor。具有與 x 相同的類型。

此操作計算索引排列的逆。它采用一維整數張量 x ,它表示從零開始的數組的索引,並將每個值與其索引位置交換。換句話說,對於輸出張量 y 和輸入張量 x ,此操作計算以下內容:

y[x[i]] = i for i in [0, 1, ..., len(x) - 1]

值必須包含 0。不能有重複值或負值。

例如:

# tensor `x` is [3, 4, 0, 2, 1]
invert_permutation(x) ==> [2, 4, 3, 0, 1]

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.invert_permutation。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。