PyTorch图算子接口与PyTorch接口一致,已优化函数列表如表1所示。
名称
说明
normalize
使用均值和标准差对张量图像进行归一化。
permute
调换张量各维度顺序,并改为在内存中连续分布的形式。
resize
调整图像大小,本文档针对bilinear和bicubic两种插值方式进行优化。