CANN/pypto 填充操作

pypto.pad

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

对输入 Tensor 进行填充（Padding）。

填充大小根据pad参数从输入 Tensor 的最后一个维度开始，由后向前依次描述。pad参数的格式为 $(pad_left, pad_right, pad_top, pad_bottom, ...)$。当前实现仅支持对最后两个维度进行常量（Constant）模式的右侧（Right）和底部（Bottom）填充。

函数原型

pad(input: Tensor, pad: Sequence[int], mode: str = "constant", value: float = 0.0) -> Tensor

参数说明

返回值说明

返回输出 Tensor，Tensor 的数据类型和input相同，Shape 为根据pad参数在对应维度上扩展后的大小。

约束说明

1. pad参数的长度必须为2或者4，pad参数中的填充大小序列的值必须为非负整数。负值填充不支持。如果传入负值，将抛出ValueError。
2. 当前仅支持多维情况下在右侧（Right）和底部（Bottom）进行填充，或者1维情况下在右侧（Right）填充。即pad序列中向左和向上的填充量必须为 0（例如格式必须为(0, pad_right, 0, pad_bottom)或者(0, pad_right)）。
3. mode当前仅支持'constant'（常量填充）模式，其他模式暂不支持。
4. value 支持任意浮点数值，填充值的数据类型会自动转换为与输入 Tensor 一致。
5. 如果input不是 Tensor 类型，或pad不是整数序列，将抛出TypeError。

调用示例

TileShape设置示例

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

TileShape 维度应和输出一致。

示例1：输入inputshape 为[m, n]，如果对其在 n 轴右侧填充了p，则输出 shape 为[m, n+p]，TileShape 设置为[m1, n1]，则m1,n1分别用于切分输出的m,n+p轴。

pypto.set_vec_tile_shapes(4, 16)

接口调用示例

# 示例：对一个 shape 为 [1, 1, 2, 2] 的 Tensor 进行填充 # 最后一个维度 (右侧) 填充 1 # 倒数第二个维度 (底部) 填充 1 t4d = pypto.tensor([0.0, 1.0, 2.0, 3.0], pypto.DT_FP32) # 假设内部已将一维数据 reshape 为 [1, 1, 2, 2] p1 = (0, 1, 0, 1) # (pad_left=0, pad_right=1, pad_top=0, pad_bottom=1) out = pypto.pad(t4d, p1, mode="constant", value=0.0)

结果示例如下：

# 输入数据 t4d (逻辑 shape 为 [1, 1, 2, 2]): [[[[0.0, 1.0], [2.0, 3.0]]]] # 输出数据 out (逻辑 shape 扩展为 [1, 1, 3, 3]): [[[[0.0, 1.0, 0.0], [2.0, 3.0, 0.0], [0.0, 0.0, 0.0]]]]

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto