pypto.log
【免费下载链接】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 推理系列产品 | √ |
功能说明
对input做以e为底的对数运算
接口原型
pypto.log(input, precision_type=pypto.LogAlgorithm.INTRINSIC) -> Tensor参数说明
| 参数 | 类型 | 说明 |
|---|---|---|
| input | Tensor | 源操作数。 支持的类型为:Tensor。 Tensor支持的数据类型为:DT_FP32, DT_FP16, DT_BF16。 支持的维度:1-4维 不支持空Tensor;Shape Size不大于2147483647(即INT32_MAX)。 |
| precision_type | LogAlgorithm, 可选 | 对数操作的精度模式。默认值为LogAlgorithm.INTRINSIC。INTRINSIC:直接使用芯片指令进行计算,速度更快。 HIGH_PRECISION:使用更高精度的计算方式,减少精度损失。 |
返回值说明
返回输出Tensor,Tensor的数据类型和input相同,Shape为input大小。
调用示例
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([3], pypto.DT_FP32) y = pypto.log(x)结果示例如下:
输入数据x: [1.0 2.0 3.0] 输出数据y: [[0.0000 0.6931 1.0986]高精度模式示例
x = pypto.tensor([3], pypto.DT_FP16) y = pypto.log(x, pypto.LogAlgorithm.HIGH_PRECISION)指令模式示例
x = pypto.tensor([3], pypto.DT_FP16) y = pypto.log(x, pypto.LogAlgorithm.INTRINSIC)【免费下载链接】pyptoPyPTO(发音: pai p-t-o):Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考