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


Python tf.raw_ops.UnravelIndex用法及代碼示例


將平麵索引數組轉換為坐標數組元組。

用法

tf.raw_ops.UnravelIndex(
    indices, dims, name=None
)

參數

  • indices 一個Tensor。必須是以下類型之一:int32 , int64。一個 0-D 或 1-D int 張量,其元素是維度數組的扁平版本的索引。
  • dims 一個Tensor。必須與 indices 具有相同的類型。一維int張量。用於解開索引的數組的形狀。
  • name 操作的名稱(可選)。

返回

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

例子:

y = tf.unravel_index(indices=[2, 5, 7], dims=[3, 3])
# 'dims' represent a hypothetical (3, 3) tensor of indices:
# [[0, 1, *2*],
#  [3, 4, *5*],
#  [6, *7*, 8]]
# For each entry from 'indices', this operation returns
# its coordinates (marked with '*'), such as
# 2 ==> (0, 2)
# 5 ==> (1, 2)
# 7 ==> (2, 1)
y ==> [[0, 1, 2], [2, 2, 1]]

numpy 兼容性

相當於 np.unravel_index

相關用法


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