• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# ReduceFusion
2
3
4## 概述
5
6减小x张量的维度。
7
8该算子对应的[NodeType](_n_n_rt.md#nodetype)为NODE_TYPE_REDUCE_FUSION。
9
10参数说明:mode若为REDUCE_ALL,REDUCE_PROD和REDUCE_MEAN,则reduce_to_end为true时,Reduce之后的输出乘以coeff为最终输出。
11
12输入:
13
14- x,n维tensor,n<8。
15
16- axis,1维tensor,指定reduce的维度,axis中每个元素的取值范围为[-n,n)。
17
18输出:
19
20- output,执行Reduce之后的m维的tensor,其数据类型和x相同。当keepDims为false时,m<n;当keepDims为true时,m==n。
21
22**Since:**
23
243.2
25
26**Version:**
27
281.0
29
30**相关模块:**
31
32[NNRt](_n_n_rt.md)
33
34
35## 汇总
36
37
38### Public 属性
39
40  | 名称 | 描述 |
41| -------- | -------- |
42| [keepDims](#keepdims) | boolean<br/>维度是否保持不变。 |
43| [mode](#mode) | enum&nbsp;[ReduceMode](_n_n_rt.md#reducemode)<br/>减小张量维度的算法,详情请参考:[ReduceMode](_n_n_rt.md#reducemode)。 |
44| [reduceToEnd](#reducetoend) | boolean<br/>如果为true,则从axis取第一个元素并设置为i,然后axis会被修改为[i,i+1,...,n-1,n]。 |
45| [coeff](#coeff) | float<br/>系数 |
46
47
48## 类成员变量说明
49
50
51### coeff
52
53
54```
55float ReduceFusion::coeff
56```
57
58**描述:**
59
60系数
61
62
63### keepDims
64
65
66```
67boolean ReduceFusion::keepDims
68```
69
70**描述:**
71
72维度是否保持不变。
73
74
75### mode
76
77
78```
79enum ReduceMode ReduceFusion::mode
80```
81
82**描述:**
83
84减小张量维度的算法,详情请参考:[ReduceMode](_n_n_rt.md#reducemode)。
85
86
87### reduceToEnd
88
89
90```
91boolean ReduceFusion::reduceToEnd
92```
93
94**描述:**
95
96如果为true,则从axis取第一个元素并设置为i,然后axis会被修改为[i,i+1,...,n-1,n],例如reduceToEnd=True,axis=[2,4],x的维度为7,则axis会被修改为[2,3,4,5,6]。
97