插值 ¶

class torch.ao.nn.quantized.functional.interpolate(input, size=None, scale_factor=None, mode='nearest', align_corners=None)

对输入进行下/上采样，使其变为给定的 size 或给定的 scale_factor

请参阅 torch.nn.functional.interpolate() 以获取实现细节。

输入维度以以下形式解释：小批量 x 通道 x [可选深度] x [可选高度] x 宽度。

注意

输入量化参数会传播到输出。

注意

仅支持 2D/3D 输入的量化输入。

注意

仅支持以下模式进行量化输入：

• 双线性插值。

• 最近

参数:

• 输入（张量）- 输入张量

• 大小（int 或 Tuple[int] 或 Tuple[int, int] 或 Tuple[int, int, int]）- 输出空间大小。

• scale_factor（float 或 Tuple[float]）- 空间大小的乘数。如果是一个元组，则必须与输入大小匹配。

• mode（str）- 用于上采样的算法： 'nearest' | 'bilinear'

• align_corners（布尔值，可选）- 几何上，我们将输入和输出的像素视为正方形而不是点。如果设置为 True ，则输入和输出张量通过其角像素的中心点对齐，保留角像素的值。如果设置为 False ，则输入和输出张量通过其角像素的角点对齐，并且插值使用边界值填充超出边界值，当 scale_factor 保持不变时，此操作与输入大小无关。这仅在 mode 为 'bilinear' 时才有影响。默认： False