TensorFlow函数:tf.unsorted_segment_max

tf.unsorted_segment_max函数

tf.unsorted_segment_max(
    data,
    segment_ids,
    num_segments,
    name=None
)

沿着张量的片段计算最大值.

此运算符与未排序的段总和运算符类似.它不是计算段的总和,而是计算最大值:

\(output_i = \ max_j data_j \),其中 max 超过了 j 这个值,segment_ids[j] == i.

如果给定分段 ID i 的最大值为空,则它会输出特定数值类型的最小可能值,output[i] = numeric_limits<T>::min().

TensorFlow函数

函数参数:

  • data:一个 Tensor,必须是下列类型之一:float32,float64,int32,uint8,int16,int8,int64,bfloat16,uint16,half,uint32,uint64.
  • segment_ids:一个 Tensor,必须是以下类型之一:int32,int64;一维张量,其秩等于 data 第一维的秩.
  • num_segments:一个 Tensor,必须是以下类型之一:int32,int64.
  • name:操作的名称(可选).

函数返回值:

tf.unsorted_segment_max函数返回一个Tensor,它与 data 的类型相同.

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

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部