CANN/pypto填充操作API文档

pypto.fillpad

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

产品支持情况

功能说明

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

和pad不同，此接口不会改变张量的形状，他将填充区域(即超过validshape的区域)用指令的值进行填充。当前实现支持输入1-2维tensor， 进行常量（Constant）模式的右侧（Right）和底部（Bottom）填充。

函数原型

fillpad(input: Tensor, mode: str = "constant", value: float = 0.0) -> Tensor

参数说明

返回值说明

返回输出 Tensor，Tensor 的数据类型和input相同，Shape 也和input相同。

约束说明

1. mode当前仅支持'constant'（常量填充）模式，其他模式暂不支持。
2. value 支持任意浮点数值，填充值的数据类型会自动转换为与输入 Tensor 一致。
3. 如果input不是 Tensor 类型，将抛出TypeError。

调用示例

TileShape设置示例

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

TileShape 维度应和输出一致。

示例1：输入inputshape 为[m, n]，则输出 shape 为[m, n]，TileShape 设置为[m1, n1]，则m1,n1分别用于切分输出的m,n轴。

pypto.set_vec_tile_shapes(4, 16)

接口调用示例

a = pypto.tensor([4, 4], pypto.DT_FP32) out = pypto.fillpad(a, "constant", "-inf")

结果示例如下：

# 输入数据 t4d (逻辑 shape 为 [4, 4]): [[1.0, 2.0, 0.0, 0.0], [3.0, 4.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0]] # 输出数据 out (逻辑 shape 为 [4, 4]): [[1.0, 2.0, -inf, -inf], [3.0, 4.0, -inf, -inf], [-inf, -inf, -inf, -inf], [-inf, -inf, -inf, -inf]]

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