_reduce_fusion.md 1.7 KB
Newer Older
Z
z00524957 已提交
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
# ReduceFusion


## 概述

减小x张量的维度。

该算子对应的[NodeType](_n_n_rt.md#nodetype)为NODE_TYPE_REDUCE_FUSION。

参数说明:mode若为REDUCE_ALL,REDUCE_PROD和REDUCE_MEAN,则reduce_to_end为true时,Reduce之后的输出乘以coeff为最终输出。

输入:

- x,n维tensor,n<8。

- axis,1维tensor,指定reduce的维度,axis中每个元素的取值范围为[-n,n)。

输出:

- output,执行Reduce之后的m维的tensor,其数据类型和x相同。当keepDims为false时,m<n;当keepDims为true时,m==n。

**Since:**

3.2

**Version:**

1.0

**相关模块:**

[NNRt](_n_n_rt.md)


## 汇总


### Public 属性

  | 名称 | 描述 | 
| -------- | -------- |
| [keepDims](#keepdims) | boolean<br/>维度是否保持不变。 | 
| [mode](#mode) | enum&nbsp;[ReduceMode](_n_n_rt.md#reducemode)<br/>减小张量维度的算法,详情请参考:[ReduceMode](_n_n_rt.md#reducemode)。 | 
| [reduceToEnd](#reducetoend) | boolean<br/>如果为true,则从axis取第一个元素并设置为i,然后axis会被修改为[i,i+1,...,n-1,n]。 | 
| [coeff](#coeff) | float<br/>系数 | 


## 类成员变量说明


### coeff

  
```
float ReduceFusion::coeff
```

**描述:**

系数


### keepDims

  
```
boolean ReduceFusion::keepDims
```

**描述:**

维度是否保持不变。


### mode

  
```
enum ReduceMode ReduceFusion::mode
```

**描述:**

减小张量维度的算法,详情请参考:[ReduceMode](_n_n_rt.md#reducemode)


### reduceToEnd

  
```
boolean ReduceFusion::reduceToEnd
```

**描述:**

如果为true,则从axis取第一个元素并设置为i,然后axis会被修改为[i,i+1,...,n-1,n],例如reduceToEnd=True,axis=[2,4],x的维度为7,则axis会被修改为[2,3,4,5,6]。