Skip to content

Latest commit

 

History

History
34 lines (21 loc) · 1.12 KB

segment_min_cn.rst

File metadata and controls

34 lines (21 loc) · 1.12 KB
Raw

segment_min

.. py:function:: paddle.geometric.segment_min(data, segment_ids, name=None)

分段求最小值函数。

此运算符，将 segment_ids 中相同索引对应的 data 的元素，进行求最小值操作。其中 segment_ids 是一个单调非减序列。 具体而言，该算子计算一个 Tensor out，使得

out_i = \min_{j \in \{segment\_ids_j == i \} } data_{j}

其中求最小值的索引 j，是符合 segment_ids[j] == i 的所有 j 。

参数

  • data (Tensor) - Tensor，数据类型为 float32、float64。
  • segment_ids (Tensor) - 一维 Tensor，与输入数据 data 的第一维大小相同，表示 data 分段位置，单调非减。合法的数据类型为 int32、int64。
  • name (str，可选) - 具体用法请参见 :ref:`api_guide_Name`，一般无需设置，默认值为 None。

返回

Tensor，分段求最小值的结果。空的 segment_id 对应的默认值为 0。

代码示例

COPY-FROM: paddle.geometric.segment_min