blob: 35f55fe1063a56650bdd83dce3599595a3bad766 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
op {
graph_op_name: "SegmentMax"
in_arg {
name: "segment_ids"
description: <<END
A 1-D tensor whose rank is equal to the rank of `data`'s
first dimension. Values should be sorted and can be repeated.
END
}
out_arg {
name: "output"
description: <<END
Has same shape as data, except for dimension 0 which
has size `k`, the number of segments.
END
}
summary: "Computes the maximum along segments of a tensor."
description: <<END
Read
[the section on segmentation](https://tensorflow.org/api_guides/python/math_ops#Segmentation)
for an explanation of segments.
Computes a tensor such that
\\(output_i = \max_j(data_j)\\) where `max` is over `j` such
that `segment_ids[j] == i`.
If the max is empty for a given segment ID `i`, `output[i] = 0`.
<div style="width:70%; margin:auto; margin-bottom:10px; margin-top:20px;">
<img style="width:100%" src="https://www.tensorflow.org/images/SegmentMax.png" alt>
</div>
END
}
|
