CANN/pypto填充操作API

pypto.pad【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明对输入 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参数说明参数名输入/输出说明input输入需要进行填充的源操作数。支持的类型为Tensor。Tensor支持的数据类型为DT_FP32、DT_FP16、DT_BF16。不支持空TensorShape仅支持1-4维Shape Size不大于2147483647即INT32_MAX。pad输入填充大小序列。支持的类型为tuple 或 list (包含int)。序列长度 $m$ 必须为偶数且满足 $\frac{m}{2} \leq$input的维度数。格式为(pad_left, pad_right, pad_top, pad_bottom, ...)所有填充大小序列的值必须为非负整数负值不支持。mode输入填充模式。支持的类型为str。可选值为constant、reflect、replicate或circular。默认值constant。注意当前仅支持constant模式。value输入当填充模式为常量填充 (constant) 时的填充值。支持的类型为float。支持任意浮点数值包括-inf、inf、0.0以及其他任意浮点数如1.0、-1.0、0.5等。 默认值0.0。返回值说明返回输出 TensorTensor 的数据类型和input相同Shape 为根据pad参数在对应维度上扩展后的大小。约束说明pad参数的长度必须为2或者4pad参数中的填充大小序列的值必须为非负整数。负值填充不支持。如果传入负值将抛出ValueError。当前仅支持多维情况下在右侧Right和底部Bottom进行填充或者1维情况下在右侧Right填充。即pad序列中向左和向上的填充量必须为 0例如格式必须为(0, pad_right, 0, pad_bottom)或者(0, pad_right)。mode当前仅支持constant常量填充模式其他模式暂不支持。value 支持任意浮点数值填充值的数据类型会自动转换为与输入 Tensor 一致。如果input不是 Tensor 类型或pad不是整数序列将抛出TypeError。调用示例TileShape设置示例说明调用该 operation 接口前应通过set_vec_tile_shapes设置 TileShape。TileShape 维度应和输出一致。示例1输入inputshape 为[m, n]如果对其在 n 轴右侧填充了p则输出 shape 为[m, np]TileShape 设置为[m1, n1]则m1,n1分别用于切分输出的m,np轴。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_left0, pad_right1, pad_top0, pad_bottom1) out pypto.pad(t4d, p1, modeconstant, value0.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-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考