接口说明

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

表1 PyTorch图算子已优化函数

名称

说明

normalize

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

permute

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

resize

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