Функция PyTorch torch.quantized_max_pool2d

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.quantized_max_pool2dЭто функция в PyTorch, используемая для выполнения операций двумерного максимального пулинга над квантованными тензорами. Эта функция обычно используется для пространственной понижающей дискретизации в квантованных сверточных нейронных сетях.

Определение функции

torch.quantized_max_pool2d(input, kernel_size, stride, padding, dilation)

Описание параметра

  • input: Входной квантованный тензор (4D: batch x channel x height x width)
  • kernel_size: Размер окна пулинга
  • stride: Stride (необязательно)
  • padding: Padding (необязательно)
  • dilation: Dilation (необязательно)

Пример использования

Пример

import torch

# Создать входной квантованный тензор (batch=1, channel=1, height=4, width=4)
input = torch.quantize_per_tensor(torch.randn(1, 1, 4, 4), scale=0.1, zero_point=0, dtype=torch.quint8)

# Выполнить квантованный максимальный пулинг (окно пулинга 2x2)
output = torch.quantized_max_pool2d(input, kernel_size=2, stride=2)

print("Форма входных данных:", input.shape)
print("Форма выхода:", output.shape)

Результат вывода:

输入形状: torch.Size([1, 1, 4, 4])
输出形状: torch.Size([1, 1, 2, 2])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения