鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

接口说明

PyTorch图算子库接口与PyTorch接口一致,已优化函数列表如表1所示。

表1 PyTorch图算子库已优化函数

名称

说明

normalize

使用均值和标准差对张量图像进行归一化。

permute

调换张量各维度顺序,并改为在内存中连续分布的形式。

resize

调整图像大小: 本文档针对bilinear和bicubic两种插值方式进行优化。