PyTorch torch.quantized_max_pool1d 函数
Pytorch torch 参考手册torch.quantized_max_pool1d 是 PyTorch 中用于对量化张量执行一维最大池化操作的函数。该函数在量化卷积神经网络中用于下采样。
函数定义
torch.quantized_max_pool1d(input, kernel_size, stride, padding, dilation)
参数说明
input: 输入的量化张量（3D: batch x channel x length）kernel_size: 池化窗口大小stride: 步长（可选）padding: 填充（可选）dilation: 膨胀（可选）
使用示例
实例
import torch
# 创建量化输入张量 (batch=1, channel=1, length=10)
input = torch.quantize_per_tensor(torch.randn(1, 1, 10), scale=0.1, zero_point=0, dtype=torch.quint8)
# 执行量化最大池化
output = torch.quantized_max_pool1d(input, kernel_size=3, stride=2, padding=1)
print("输入形状:", input.shape)
print("输出形状:", output.shape)
# 创建量化输入张量 (batch=1, channel=1, length=10)
input = torch.quantize_per_tensor(torch.randn(1, 1, 10), scale=0.1, zero_point=0, dtype=torch.quint8)
# 执行量化最大池化
output = torch.quantized_max_pool1d(input, kernel_size=3, stride=2, padding=1)
print("输入形状:", input.shape)
print("输出形状:", output.shape)
输出结果为：
输入形状: torch.Size([1, 1, 10]) 输出形状: torch.Size([1, 1, 5])