pypto.gather

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

功能说明

根据PatternMode选择的内置Mask，对输入Tensor对应Bit位为1的位置组成输出Tensor，Bit位为0的值直接丢弃。PatternMode有7种模式：

• PatternMode=1，尾轴每两个元素取第一个元素

• PatternMode=2，尾轴每两个元素取第二个元素

• PatternMode=3，尾轴每四个元素取第一个元素

• PatternMode=4，尾轴每四个元素取第二个元素

• PatternMode=5，尾轴每四个元素取第三个元素

• PatternMode=6，尾轴每四个元素取第四个元素

• PatternMode=7，尾轴取全部元素

函数原型

gathermask(self: Tensor, pattern_mode: int) -> Tensor

参数说明

参数名	输入/输出	说明
self	输入	源操作数。 支持的类型为：Tensor。 Tensor支持的数据类型为：DT_INT16，DT_UINT16，DT_INT32, DT_UINT32，DT_FP16，DT_BF16, DT_FP32。 不支持空 Tensor，Shape 支持1-4维，且shape size不大于2147483647（即INT32_MAX）。
pattern_mode	输入	源操作数。 int类型 ，取值范围为：1~7。

返回值说明

返回输出 Tensor，输出Tensor数据类型与 self 数据类型一致，输出 Tensor的Shape如下：

• pattern_mode <= 2，输出shape尾轴=self.shape尾轴/2，其它轴和输入shape一致；

• 2 < pattern_mode < 7, 输出shape尾轴=self.shape尾轴/4，其它轴和输入shape一致；

• pattern_mode = 7, 输出shape=输入shape

约束说明

1. 当1 <= pattern_mode <= 2时:

   • self.shape尾轴必须能被2整除

   • tileshape尾轴必须是2的整倍数

   • viewshape尾轴必须是2的整倍数

   • self.shape尾轴不做view切分

2. 当3 <= pattern_mode <= 6时:

   • self.shape尾轴必须是4的整倍数

   • tileshape尾轴必须是4的整倍数

   • viewshape尾轴必须是4的整倍数

   • self.shape尾轴不做view切分

调用示例

TileShape设置示例

说明：调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输出一致。

如输入self为[x, y, z]，pattern_mode为1，输出为[x, y, z/2]，TileShape设置为[x1, y1, 2*z1]，则x1, y1, 2*z1分别用于切分x, y, z轴。

pypto.set_vec_tile_shapes(4, 16, 32)

接口调用示例

x = pypto.tensor([3, 6], pypto.DT_INT32)        # shape (3, 6)
pattern_mode = 1
y = pypto.gathermask(x, pattern_mode)

结果示例如下：

输入数据 x: [[0,  1,  2,  3,  4,  5],
             [6,  7,  8,  9,  10,  11],
             [12,  13,  14,  15,  16,  17]]
     pattern_mode: 1
输出数据 y: [[0,  2,  4],
             [6,  8,  10],
             [12,  14,  16]]