paddle.compat
paddle.compat 目录下包含飞桨框架支持的 PyTorch 兼容函数与模块接口
PyTorch 兼容函数
API 名称 |
API 功能 |
|---|---|
|
|
包含 amax、同时返回 values 及 indices 的轴向最大值、maximum 三种功能 |
|
|
包含 amin、同时返回 values 及 indices 的轴向最小值、minimum 三种功能 |
|
|
softmax 函数 |
|
|
同时返回 values 及 indices 的排序 |
|
|
允许非整除块大小输入的 Tensor 轴向切分 |
|
|
兼容版中位数，支持 dim/keepdim/out 签名 |
|
|
忽略 NaN 的兼容版中位数，支持 dim/keepdim/out 签名 |