harmony 鸿蒙SpaceToBatchND
SpaceToBatchND
概述
将4维张量在空间维度上进行切分成多个小块，然后在batch维度上拼接这些小块。
该算子对应的NodeType为NODE_TYPE_SPACE_TO_BATCH_ND。
输入：
- x，n维tensor
输出： - output：一个4维tensor，数据类型和input一致。shape由input、blockShape和paddings共同决定，假设input shape为[n,c,h,w]，则有：
$ output.shape[0] = n * blockShape[0] * blockShape[1] $
$ output.shape[1] = c $
$ output.shape[2] = (h + paddings[0][0] + paddings[0][1]) / blockShape[0] $
$ output.shape[3] = (w + paddings[1][0] + paddings[1][1]) / blockShape[1] $
要求$ (h + paddings[0][0] + paddings[0][1]) $能被$ blockShape[0] $整除，$ (w + paddings[1][0] + paddings[1][1]) $能被$ blockShape[1] $整除。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| blockShape | long[] 描述空间维度为分割的个数，取值需大于1。 |
| paddings | long[][] 空间维度的填充大小。 |
类成员变量说明
blockShape
long [] SpaceToBatchND::blockShape
描述:
描述空间维度为分割的个数，取值需大于1。
paddings
long [][] SpaceToBatchND::paddings
描述:
空间维度的填充大小。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
7、 openharmony