• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# BatchToSpaceND
2
3
4## 概述
5
6将一个4维tensor的batch维度按block_shape切分成小块,并将这些小块拼接到空间维度。
7
8该算子对应的[NodeType](_n_n_rt.md#nodetype)为NODE_TYPE_BATCH_TO_SPACE_ND。
9
10输入:
11
12- x,n维tensor。
13
14输出:
15
16- 输出张量,假设x的形状为(n,h,w,c),output的形状为(n',h',w',c'):
17
18  - $ n' = n / (block_shape[0] \* block_shape[1]) $
19  - $ h' = h \* block_shape[0] - crops[0][0] - crops[0][1] $
20  - $ w' = w \* block_shape[1] - crops[1][0] - crops[1][1] $
21  - $ c'= c $
22
23**Since:**
24
253.2
26
27**Version:**
28
291.0
30
31**相关模块:**
32
33[NNRt](_n_n_rt.md)
34
35
36## 汇总
37
38
39### Public 属性
40
41  | 名称 | 描述 |
42| -------- | -------- |
43| [blockShape](#blockshape) | long[]<br/>一个长度为2的数组[height_block,weight_block],指定切分到空间维度上的block大小。 |
44| [crops](#crops) | long[][]<br/>一个shape为(2,2)的2维数组[[crop0_start,crop0_end],[crop1_start,crop1_end]],&nbsp;表示在output的空间维度上截掉部分元素。 |
45
46
47## 类成员变量说明
48
49
50### blockShape
51
52
53```
54long [] BatchToSpaceND::blockShape
55```
56
57**描述:**
58
59一个长度为2的数组[height_block,weight_block],指定切分到空间维度上的block大小。
60
61
62### crops
63
64
65```
66long [][] BatchToSpaceND::crops
67```
68
69**描述:**
70
71一个shape为(2,2)的2维数组[[crop0_start,crop0_end],[crop1_start,crop1_end]], 表示在output的空间维度上截掉部分元素。
72