paddle.compat

paddle.compat 目录下包含飞桨框架支持的 PyTorch 兼容函数与模块接口

PyTorch 兼容函数

API 名称

API 功能

max

包含 amax、同时返回 values 及 indices 的轴向最大值、maximum 三种功能

兼容版中位数,支持 dim/keepdim/out 签名

min

包含 amin、同时返回 values 及 indices 的轴向最小值、minimum 三种功能

忽略 NaN 的兼容版中位数,支持 dim/keepdim/out 签名

slogdet 函数

同时返回 values 及 indices 的排序

允许非整除块大小输入的 Tensor 轴向切分

PyTorch 兼容的 nn 模块

类名称

类功能

PyTorch 对齐的 paddle.nn.Linear 兼容版本

允许 Tensor 输入的 paddle.nn.Unfold 兼容版本

PyTorch 兼容的 nn.functional 模块

函数名称

函数功能

PyTorch 对齐的 paddle.nn.functional.linear 兼容版本

pad

从最后一维度开始进行 padding、padding 正确兼容转换的填充函数

softmax 函数