# pypto.pow【免费下载链接】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中每个元素的 other 次幂逐元素运算返回与输入形状相同的Tensor。函数原型pow(input: Tensor, other: Union[Tensor, int, float], precision_type: PowAlgorithm PowAlgorithm.HIGH_PRECISION) - Tensor参数说明参数名输入/输出说明input输入源操作数。支持的类型为Tensor。Tensor支持的数据类型为DT_FP16、DT_BF16、DT_FP32、DT_INT32。不支持空TensorShape仅支持1-4维支持按照单个维度广播到相同形状Shape Size不大于2147483647即INT32_MAX。other输入指数。支持的类型为Tensor、int或float。Tensor支持的数据类型为DT_FP16、DT_BF16、DT_FP32、DT_INT32。不支持空TensorShape仅支持1-4维支持按照单个维度广播到相同形状Shape Size不大于2147483647即INT32_MAX。precision_type输入精度模式枚举类型用以控制指数计算的精度模式具体定义为PowAlgorithm 。默认为 HIGH_PRECISION高精度模式。返回值说明返回一个与输入形状相同的Tensor其元素为输入Tensor对应元素的other次幂。当other为int时返回的Tensor的数据类型与输入相同。当other为float时若输入Tensor类型为DT_INT32则返回DT_FP32否则返回的Tensor的数据类型与输入相同。当other为Tensor时返回的Tensor的数据类型见数据类型提升说明章节。数据类型提升说明我们约定float32float16bfloat16int32。当两个输入参数类型一个为float16而另一个bfloat16时输出的数据类型为float32。其他情况下输出类型为输入参数类型的更大值如输入float32和float16则输出为float32。参数类型float32float16bfloat16int32float32float32float32float32float32float16float32float16float32float16bfloat16float32float32bfloat16bfloat16int32float32float16bfloat16int32调用示例TileShape设置示例说明调用该operation接口前应通过set_vec_tile_shapes设置TileShape。TileShape维度应和输出一致。示例1输入input shape为[m, n]输出为[m, n]TileShape设置为[m1, n1], 则m1, n1分别用于切分m, n轴。pypto.set_vec_tile_shapes(4, 16)接口调用示例x pypto.tensor([2, 2], pypto.DT_FP32) a 2 b pypto.tensor([2, 2], pypto.DT_FP32) y pypto.pow(x, a) z pypto.pow(x, b)结果示例如下输入数据x: [[1.0 2.0], [-3.0 4.0]] 输入数据b: [[2.0 2.0], [1.0 1.0]] 输出数据y: [[1.0 4.0], [9.0 16.0]] 输出数据z: [[1.0 4.0], [-3.0 4.0]]【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考