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

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


查找最后一个维度的 n - 阶统计值的值。

用法

tf.raw_ops.NthElement(
    input, n, reverse=False, name=None
)

参数

  • input 一个Tensor。必须是以下类型之一：float32 , float64 , int32 , uint8 , int16 , int8 , int64 , bfloat16 , uint16 , half , uint32 , uint64。一维或更高，最后一维至少 n+1 。
  • n Tensor 类型为 int32 。 0-D。要沿最后一个维度选择的排序向量的位置(沿矩阵的每一行)。 n的有效范围是[0, input.shape[:-1])
  • reverse 可选的 bool 。默认为 False 。当设置为 True 时，在向量中找到 nth-largest 值，反之亦然。
  • name 操作的名称(可选)。

返回

  • 一个Tensor。具有与 input 相同的类型。

如果输入是向量(rank-1)，则在向量中找到nth-smallest 值的条目，并将它们的值作为标量张量输出。

对于矩阵(分别是更高等级的输入)，计算每行中的 nth-smallest 值(分别是沿最后一维的向量)的条目。因此，

values.shape = input.shape[:-1]

相关用法


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