_split.md 1.2 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
# Split


## 概述

算子沿 axis 维度将x拆分成多个张量,张量数量由outputNum指定。

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

输入:

- x,n维tensor

输出:

- outputs,一组n维张量,每一个张量类型和维度相同,每个张量的类型和x一致。

**Since:**

3.2

**Version:**

1.0

**相关模块:**

[NNRt](_n_n_rt.md)


## 汇总


### Public 属性

  | 名称 | 描述 | 
| -------- | -------- |
| [outputNum](#outputnum) | long<br/>指定分割数量。 | 
| [sizeSplits](#sizesplits) | long[]<br/>指定输入的张量沿axis轴拆分后,每个张量的大小。 | 
| [axis](#axis) | long<br/>指定分割的维度。 | 


## 类成员变量说明


### axis

  
```
long Split::axis
```

**描述:**

指定分割的维度。


### outputNum

  
```
long Split::outputNum
```

**描述:**

指定分割数量。


### sizeSplits

  
```
long [] Split::sizeSplits
```

**描述:**

指定输入的张量沿 axis 轴拆分后,每个张量的大小。

如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除;如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。