插值

类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 宽度。

注意

输入的量化参数会传递到输出。

注意

仅支持量化输入的二维和三维输入

注意

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

• 双线性

• 最近的邻居

参数

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