当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.raw_ops.TopKV2用法及代码示例


查找最后一个维度的 k 最大元素的值和索引。

用法

tf.raw_ops.TopKV2(
    input, k, sorted=True, name=None
)

参数

  • input 一个Tensor。必须是以下类型之一：float32 , float64 , int32 , uint8 , int16 , int8 , int64 , bfloat16 , uint16 , half , uint32 , uint64。一维或更高，最后一维至少 k 。
  • k Tensor 类型为 int32 。 0-D。沿最后一个维度(沿矩阵的每一行)查找的顶部元素的数量。
  • sorted 可选的 bool 。默认为 True 。如果为 true，则生成的 k 元素将按值降序排序。
  • name 操作的名称(可选)。

返回

  • Tensor 对象(值、索引)的元组。
  • values 一个Tensor。具有与 input 相同的类型。
  • indices Tensor 类型为 int32 。

如果输入是向量(rank-1)，则在向量中找到 k 最大的条目，并将它们的值和索引作为向量输出。因此 values[j] 是 j - input 中最大的条目，其索引是 indices[j] 。

对于矩阵(分别是更高等级的输入)，计算每行中的顶部 k 条目(分别是沿最后一个维度的向量)。因此，

values.shape = indices.shape = input.shape[:-1] + [k]

如果两个元素相等，则首先出现lower-index 元素。

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.TopKV2。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。