TensorFlow函数:tf.unravel_index

tf.unravel_index函数

tf.unravel_index(
    indices,
    dims,
    name=None
)

将 flat 索引或 flat 索引数组转换为坐标数组的元组.

函数参数:

  • indices:一个 Tensor,必须是以下类型之一:int32,int64;一个 0-D 或 1-D int 张量,其元素是索引到维度数组的 flattened 版本中的索引.
  • dims:一个 Tensor,必须与 indices 具有相同的类型;一维 int 张量,用于分解索引的数组的形状.
  • name:操作的名称(可选).

函数返回值:

该函数返回一个与 indices 具有相同类型的 Tensor.

Numpy兼容性

相当于 np.unravel_index

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部