paddle.amp¶
paddle.amp 目录下包含飞桨框架支持的动态图自动混合精度(AMP)相关的API。具体如下:
AMP相关API¶
API名称 |
API功能 |
---|---|
|
创建AMP上下文环境 |
|
根据选定混合精度训练模式,改写神经网络参数数据类型 |
|
控制loss的缩放比例 |
开启AMP后默认转化为float16计算的相关OP¶
OP名称 |
OP功能 |
---|---|
conv2d |
卷积计算 |
matmul |
矩阵乘法 |
matmul_v2 |
矩阵乘法 |
mul |
矩阵乘法 |
开启AMP后默认使用float32计算的相关OP¶
OP名称 |
OP功能 |
---|---|
exp |
指数运算 |
square |
平方运算 |
log |
对数运算 |
mean |
取平均值 |
sum |
求和运算 |
cos_sim |
余弦相似度 |
softmax |
softmax操作 |
softmax_with_cross_entropy |
softmax交叉熵损失函数 |
sigmoid_cross_entropy_with_logits |
按元素的概率误差 |
cross_entropy |
交叉熵 |
cross_entropy2 |
交叉熵 |