interpolate

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

注意

量化输入仅支持以下模式

• bilinear

• nearest

参数

• input (Tensor) – 输入张量

• size (int 或 Tuple[int] 或 Tuple[int, int] 或 Tuple[int, int, int]) – 输出空间大小。

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

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

• align_corners (bool, optional) – 从几何角度来看，我们将输入和输出的像素视为正方形而不是点。如果设置为 True，则输入和输出张量通过其角像素的中心点对齐，从而保留角像素处的值。如果设置为 False，则输入和输出张量通过其角像素的角点对齐，并且插值对超出边界的值使用边缘值填充，当 scale_factor 保持不变时，此操作独立于输入大小。仅当 mode 为 'bilinear' 时才有效。默认值：False