插值

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输入

注意

仅支持以下模式用于量化输入：

• 双线性

• 最近

Parameters

• 输入 (张量) – 输入张量

• 大小 (整数 或 元组[整数] 或 元组[整数, 整数] 或 元组[整数, 整数, 整数]) – 输出空间大小。

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

• mode (str) – 用于上采样的算法： 'nearest' | 'bilinear'

• align_corners (bool, 可选) – 在几何上，我们将输入和输出的像素视为正方形而不是点。 如果设置为 True，输入和输出张量通过其角像素的中心点对齐，保留角像素的值。 如果设置为 False，输入和输出张量通过其角像素的角点对齐，并且插值使用边缘值填充边界外的值，使得当 scale_factor 保持不变时，此操作与输入大小无关。这仅在 mode 为 'bilinear' 时生效。 默认值：False