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


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。