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

Python tf.unravel_index用法及代碼示例

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

用法

tf.unravel_index(
    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.unravel_index。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。