pypto.fillpad
【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
对输入 Tensor 进行填充（Padding）。
和pad不同，此接口不会改变张量的形状，他将填充区域(即超过validshape的区域)用指令的值进行填充。当前实现支持输入1-2维tensor， 进行常量（Constant）模式的右侧（Right）和底部（Bottom）填充。
函数原型
fillpad(input: Tensor, mode: str = "constant", value: float = 0.0) -> Tensor参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| input | 输入 | 需要进行填充的源操作数。 支持的类型为：Tensor。 Tensor支持的数据类型为：DT_FP32、DT_FP16、DT_BF16。 不支持空Tensor；Shape支持1-2维；Shape Size不大于2147483647（即INT32_MAX）。 |
| mode | 输入 | 填充模式。 支持的类型为：str。 可选值为 'constant'、'reflect'、'replicate'或'circular'。默认值： 'constant'。注意：当前仅支持 'constant'模式。 |
| value | 输入 | 当填充模式为常量填充 ('constant') 时的填充值。支持的类型为：float。 支持任意浮点数值，包括 -inf、inf、0.0以及其他任意浮点数（如1.0、-1.0、0.5等）。 默认值：0.0。 |
返回值说明
返回输出 Tensor，Tensor 的数据类型和input相同，Shape 也和input相同。
约束说明
- mode当前仅支持
'constant'（常量填充）模式，其他模式暂不支持。 - value 支持任意浮点数值，填充值的数据类型会自动转换为与输入 Tensor 一致。
- 如果
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
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
