harmony 鸿蒙BatchToSpaceND

BatchToSpaceND

将一个4维tensor的batch维度按block_shape切分成小块，并将这些小块拼接到空间维度。

该算子对应的NodeType为NODE_TYPE_BATCH_TO_SPACE_ND。

输入：

输出：

输出张量，假设x的形状为(n,h,w,c)，output的形状为（n’,h’,w’,c’）：
- $ n’ = n / (block_shape[0] * block_shape[1]) $
- $ h’ = h * block_shape[0] - crops[0][0] - crops[0][1] $
- $ w’ = w * block_shape[1] - crops[1][0] - crops[1][1] $
- $ c’= c $

起始版本： 3.2

相关模块：NNRt

名称	描述
long[] blockShape	一个长度为2的数组[height_block，weight_block]，指定切分到空间维度上的block大小。
long[][] crops	一个shape为(2，2)的2维数组[[crop0_start，crop0_end]，[crop1_start，crop1_end]]，表示在output的空间维度上截掉部分元素。

long [] BatchToSpaceND::blockShape

描述

一个长度为2的数组[height_block，weight_block]，指定切分到空间维度上的block大小。

long [][] BatchToSpaceND::crops

描述

一个shape为(2，2)的2维数组[[crop0_start，crop0_end]，[crop1_start，crop1_end]]，表示在output的空间维度上截掉部分元素。

0 赞